chenyuntc / pytorch-book

PyTorch tutorials and fun projects including neural talk, neural style, poem writing, anime generation (《深度学习框架PyTorch:入门与实战》)
MIT License
11.99k stars 3.73k forks source link

chapter6 的main函数中加载路径是否存在问题? #215

Open LawsonAbs opened 4 years ago

LawsonAbs commented 4 years ago

第24行的代码如下: train_data = DogCat(opt.test_data_root,test=True) 这里既然是train,为何用opt.test_data_root 做数据输入?

LawsonAbs commented 4 years ago

@chenyuntc 麻烦云哥给我解释一下。

wzhings commented 4 years ago

应该就是个typo, 你直接改成 test_data = DogCat(opt.test_data_root,test=True) 就行了

LawsonAbs commented 4 years ago

应该就是个typo, 你直接改成 test_data = DogCat(opt.test_data_root,test=True) 就行了

行,tks