Python requires_grad_ Examples

Programming Language: Python

Namespace/Package Name: torch.utils.data

Method/Function: requires_grad_

Examples at hotexamples.com: 2

Python requires_grad_ - 2 examples found. These are the top rated real world Python examples of torch.utils.data.requires_grad_ extracted from open source projects. You can rate examples to help us improve the quality of examples.

Example #1

Show file

File: image_helper.py Project: wangyongkang-xie/Theroy_DetectAcc

 def get_batch(self, train_data, bptt, evaluation=False):
     data, target = bptt
     data = data.to(device)
     target = target.to(device)
     if evaluation:
         data.requires_grad_(False)
         target.requires_grad_(False)
     return data, target

Example #2

Show file

File: loan_helper.py Project: nickcooper666/backdoor_loan

 def get_batch(self, train_data, bptt, evaluation=False):
     data, target = bptt
     data = data.float().cuda()
     target = target.long().cuda()
     if evaluation:
         data.requires_grad_(False)
         target.requires_grad_(False)
     return data, target