(self)
| 134 | return loss_logs |
| 135 | |
| 136 | def update(self): |
| 137 | self.zero_grad([self.opt_encoder]) |
| 138 | loss_logs = self.backward_G() |
| 139 | self.loss_mot_rec.backward() |
| 140 | self.clip_norm([self.encoder]) |
| 141 | self.step([self.opt_encoder]) |
| 142 | |
| 143 | return loss_logs |
| 144 | |
| 145 | def to(self, device): |
| 146 | if self.opt.is_train: |
no test coverage detected