diff options
| author | junyanz <junyanzhu89@gmail.com> | 2018-02-09 18:07:05 -0500 |
|---|---|---|
| committer | junyanz <junyanzhu89@gmail.com> | 2018-02-09 18:07:05 -0500 |
| commit | 4712e422bd5cfee7b5eac45a32b87a01c75b6ddd (patch) | |
| tree | c4e448bfd204474423ad12e3b0026d79f70abd99 /models/cycle_gan_model.py | |
| parent | b40e20b8d29450efb10a8e84b267d13734ae0c13 (diff) | |
cuda(...) inplace operation
Diffstat (limited to 'models/cycle_gan_model.py')
| -rw-r--r-- | models/cycle_gan_model.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/models/cycle_gan_model.py b/models/cycle_gan_model.py index 730d077..bd21d28 100644 --- a/models/cycle_gan_model.py +++ b/models/cycle_gan_model.py @@ -72,8 +72,8 @@ class CycleGANModel(BaseModel): input_A = input['A' if AtoB else 'B'] input_B = input['B' if AtoB else 'A'] if len(self.gpu_ids) > 0: - input_A = input_A.cuda(self.gpu_ids[0], async=True) - input_B = input_B.cuda(self.gpu_ids[0], async=True) + input_A.cuda(self.gpu_ids[0], async=True) + input_B.cuda(self.gpu_ids[0], async=True) self.input_A = input_A self.input_B = input_B self.image_paths = input['A_paths' if AtoB else 'B_paths'] |