When I train the resnet-18 model in pytorch imagenent example
https://github.com/pytorch/examples/blob/master/imagenet/main.py 47
there are two lines
import torch.backends.cudnn as cudnn
cudnn.benchmark = True
As I run the program, the server with 4 gtx 1080 gpus automatically reboot after several seconds. And after I set the value to False, everything is fine.
So what’s the mean of those codes and what happens when the value is set to True?