代码原料:李森科在zhihu:2.1 DataLoader与DataSet 里面的人民币二分类代码。
讨论对象:对train.py第84行进行断点调试。
代码逻辑:train_loader调用第58行,第58行调用my_dataset.py里面的train_data类和PyTorch内置函数DataLoader。
我们分别在PyCharm和VSCode中运行train.py。
1 VSCode Step Into = PyCharm Step Into My Code
先看PyCharm的调试过程:
在点击Step Into按钮后,我们进入PyTorch内置函数了。
如果点击Step Into My Code,我们进入my_dataset.py。