I used program cord of machine learning by tensorflow and pytroch. I finished program based on tensorflow. I tried to create cord of prediction based on pytorch. But, I conf