diff --git a/PyTorch/built-in/cv/classification/DenseNet161_ID0455_for_PyTorch/references/classification/train.py b/PyTorch/built-in/cv/classification/DenseNet161_ID0455_for_PyTorch/references/classification/train.py index 11c3a0450449f7b82eb1c96e223df0bf10b6e675..d484fde01350fff108e06953c8298346b4e5b42a 100644 --- a/PyTorch/built-in/cv/classification/DenseNet161_ID0455_for_PyTorch/references/classification/train.py +++ b/PyTorch/built-in/cv/classification/DenseNet161_ID0455_for_PyTorch/references/classification/train.py @@ -214,7 +214,7 @@ def main(args): if args.is_master_node: print("Creating model") - print(torchvision.models.__dict__) + # print(torchvision.models.__dict__) model = torchvision.models.__dict__[args.model](num_classes=args.num_classes) if args.pretrained: checkpoint = torch.load(args.pretrained_weight_path, map_location='cpu')