update network by back propagation
(self, loss_dict)
| 176 | self.net.to(self.device) |
| 177 | |
| 178 | def update_network(self, loss_dict): |
| 179 | """update network by back propagation""" |
| 180 | loss = sum(loss_dict.values()) |
| 181 | self.optimizer.zero_grad() |
| 182 | loss.backward() |
| 183 | self.optimizer.step() |
| 184 | self.scheduler.step() |
| 185 | |
| 186 | def _forward_batch(self, data): |
| 187 | """forward a batch of data""" |