Added torch.no_grad()
scope in PytorchEstimator.predict_proba
#61
Job | Run time |
---|---|
2m 18s | |
2m 11s | |
4m 29s |
torch.no_grad()
scope in PytorchEstimator.predict_proba
#61
Job | Run time |
---|---|
2m 18s | |
2m 11s | |
4m 29s |