Python CudaModule Examples

Example #1

 def cuda(self, device=None):
     CudaModule.cuda(self, device)
     self.sentence_embedding.cuda(device)
     self.unet.cuda(device)
     if self.ground_loss:
         self.lang_filter.cuda(device)
     return self

Example #2

File: module_with_auxiliaries_base.py Project: dxsun/drif

 def cuda(self, device=None):
     CudaModule.cuda(self, device)
     for key, aux in self.auxiliaries.items():
         aux.cuda(device)

Example #3

File: auxiliary_objective_base.py Project: hyzcn/drif

 def cuda(self, device=None):
     CudaModule.cuda(self, device)
     return self

Example #4

 def cuda(self, device=None):
     CudaModule.cuda(self, device)
     self.img_to_features.cuda(device)

Example #5

 def cuda(self, device=None):
     CudaModule.cuda(self, device)
     self.feature_net.cuda()

Example #6

 def cuda(self, device=None):
     CudaModule.cuda(self, device)
     self.stage1_visitation_prediction.cuda(device)
     self.map_transformer_w_to_r.cuda(device)
     return self

Example #7

 def cuda(self, device=None):
     CudaModule.cuda(self, device)
     self.map_affine.cuda(device)
     return self

Example #8

File: ratio_path_predictor_bak.py Project: pianpwk/drif

 def cuda(self, device=None):
     CudaModule.cuda(self, device)
     #self.map_filter.cuda(device)
     self.softmax.cuda(device)
     self.dbg_t = None
     return self

Example #9

 def cuda(self, device=None):
     CudaModule.cuda(self, device)
     self.softmax.cuda(device)
     self.logsoftmax.cuda(device)

Example #10

File: action_loss.py Project: dxsun/drif

 def cuda(self, device=None):
     CudaModule.cuda(self, device)
     self.act_loss.cuda(device)
     self.stoploss.cuda(device)