From 5f636af630436588ceec8abcd39bfe23789684fc Mon Sep 17 00:00:00 2001 From: Takyoung Kim Date: Sun, 31 Jan 2021 16:24:16 +0900 Subject: [PATCH] logging typo fix net2 model name --- train_dml.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train_dml.py b/train_dml.py index 34504c1..4b82c85 100644 --- a/train_dml.py +++ b/train_dml.py @@ -84,7 +84,7 @@ def main(): net2 = define_tsnet(name=args.net2_name, num_class=args.num_class, cuda=args.cuda) checkpoint = torch.load(args.net2_init) load_pretrained_model(net2, checkpoint['net']) - logging.info('Net2: %s', net1) + logging.info('Net2: %s', net2) logging.info('Net2 param size = %fMB', count_parameters_in_MB(net2)) logging.info('-----------------------------------------------') @@ -344,4 +344,4 @@ def adjust_lr(optimizers, epoch): if __name__ == '__main__': - main() \ No newline at end of file + main()