Python requires_grad_示例

示例#1

文件： image_helper.py 项目： 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

示例#2

文件： loan_helper.py 项目： 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