diff --git a/mu_map/training/default.py b/mu_map/training/default.py index 90ab158e2fec846cb4ef7da077a1e527dafc0744..6ef94d9d2e0b29213cb9fdd6cf950866ee54f892 100644 --- a/mu_map/training/default.py +++ b/mu_map/training/default.py @@ -205,6 +205,8 @@ if __name__ == "__main__": device = torch.device(args.device) logger = get_logger_by_args(args) + logger.info(args) + model = UNet(in_channels=1, features=args.features) model = model.to(device)