hskha21/Brain-Stroke-Diagnosis

Brain-Stroke-Diagnosis

Notes for Pytorch implementation

• —optimizer.zerograd(settonone=True) is faster than optimizer.zerograd() for large models
• —torch.inferencemode() is better than torch.nograd() for inference
• —torch.autograd.setdetectanomaly(True) is useful for debugging