search

AlexHex7 / Non-local_pytorch

Implementation of Non-local Block.
Apache License 2.0
1.57k stars 277 forks source link

Update demo_MNIST_AMP_train_with_single_gpu.py #49

Closed AlexHex7 closed 2 years ago

AlexHex7 commented 2 years ago

Remove redundant code 'net.train()'