Open gavrin-s opened 5 years ago
Hi! Your test_on_images.ipynb is not reproduced. You need to fix softmax(a, dim=-1) to softmax(a, dim=1) in get_nets.py everywhere.
test_on_images.ipynb
softmax(a, dim=-1)
softmax(a, dim=1)
get_nets.py
Hi! Your
test_on_images.ipynbis not reproduced. You need to fixsoftmax(a, dim=-1)tosoftmax(a, dim=1)inget_nets.pyeverywhere.