train.py:21 epoch/all_epoch

ChawDoe / LeNet5-MNIST-PyTorch

The simplest implementation of LeNet5 with mnist in PyTorch. Accuracy: ~99%

MIT License

193 stars 56 forks source link

Closed binotation closed 2 years ago

binotation commented 2 years ago

train.py:21 is epoch supposed to be all_epoch? Otherwise, epoch is undefined.

binotation commented 2 years ago

epoch 需要改成 all_epoch?

ChawDoe commented 2 years ago

Right. Thank you