Run MP-CNN on torch-0.3.0.post4
Closed this issue · 2 comments
hinhmd commented
Training speed has increased significantly with the new version of Pytorch (4x to 256x faster Softmax/LogSoftmax).
Please consider experimenting with the new version.
tuzhucheng commented
Yeah I'll update to PyTorch v0.3.0 and re-tune hyperparameters.
tuzhucheng commented
Upgraded to PyTorch 0.3.0, however, performance on some datasets dropped. Using #5 to keep track.