(self, model, args)
| 24 | class Architect(object): |
| 25 | |
| 26 | def __init__(self, model, args): |
| 27 | self.network_weight_decay = args.wdecay |
| 28 | self.network_clip = args.clip |
| 29 | self.model = model |
| 30 | self.optimizer = torch.optim.Adam(self.model.arch_parameters(), lr=args.arch_lr, weight_decay=args.arch_wdecay) |
| 31 | |
| 32 | def _compute_unrolled_model(self, hidden, input, target, eta): |
| 33 | loss, hidden_next = self.model._loss(hidden, input, target) |
nothing calls this directly
no test coverage detected