Add save/load_best for xlayers

2021-05-13 07:57:41 +00:00
parent a2b1d0d227
commit d1836cbe52
4 changed files with 73 additions and 38 deletions
--- a/exps/LFNA/lfna-tall-hpnet.py
+++ b/exps/LFNA/lfna-tall-hpnet.py
@@ -36,7 +36,7 @@ def main(args):
    model = get_model(**model_kwargs)
    criterion = torch.nn.MSELoss()

-    logger.log("There are {:} weights.".format(model.get_w_container().numel()))
+    logger.log("There are {:} weights.".format(model.numel()))

    shape_container = model.get_w_container().to_shape_container()
    hypernet = HyperNet(shape_container, args.hidden_dim, args.task_dim)