u39kun / deep-learning-benchmark

Deep Learning Benchmark for comparing the performance of DL frameworks, GPUs, and single vs half precision
429 stars 65 forks source link

Update models.py #14

Closed dd52 closed 1 year ago

dd52 commented 4 years ago

deactivating autograd: changed deprecated volatile=True command to with torch.no_grad() block