"tests/nn/checkpoint/test_checkpoint_activations.py" did not exist on "e92e85ceb5bb1352e2b4a0f645275f181a159018"
-
Jiezhong Qiu authored
* when k=1, it reduces to torch.max, and not surprising that torch.max is faster than torch.topk. * however when k=2, it is even slower than torch.topk
1feaaf0c
