Print out the network information.
(self, model, name)
| 87 | self.D.to(self.device) |
| 88 | |
| 89 | def print_network(self, model, name): |
| 90 | """Print out the network information.""" |
| 91 | num_params = 0 |
| 92 | for p in model.parameters(): |
| 93 | num_params += p.numel() |
| 94 | print(model) |
| 95 | print(name) |
| 96 | print("The number of parameters: {}".format(num_params)) |
| 97 | |
| 98 | def restore_model(self, resume_iters): |
| 99 | """Restore the trained generator and discriminator.""" |