Added torch.no_grad()
scope in PytorchEstimator.predict_proba
#62
Job | Run time |
---|---|
2m 6s | |
2m 20s | |
4m 26s |
torch.no_grad()
scope in PytorchEstimator.predict_proba
#62
Job | Run time |
---|---|
2m 6s | |
2m 20s | |
4m 26s |