Skip to content
Snippets Groups Projects
Commit 58c38ae5 authored by Tamino Huxohl's avatar Tamino Huxohl
Browse files

default training now logs args

parent 686d2316
No related branches found
No related tags found
No related merge requests found
...@@ -205,6 +205,8 @@ if __name__ == "__main__": ...@@ -205,6 +205,8 @@ if __name__ == "__main__":
device = torch.device(args.device) device = torch.device(args.device)
logger = get_logger_by_args(args) logger = get_logger_by_args(args)
logger.info(args)
model = UNet(in_channels=1, features=args.features) model = UNet(in_channels=1, features=args.features)
model = model.to(device) model = model.to(device)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment