From 58b0b7c1a560c38680b846ce7a899ec9eeb310e0 Mon Sep 17 00:00:00 2001 From: Ryan Date: Thu, 26 May 2022 02:45:18 +0000 Subject: [PATCH] =?UTF-8?q?update=20DenseNet161=5FID0455=5Ffor=5FPyTorch/r?= =?UTF-8?q?eferences/classification/train.py.=20=E5=BD=B1=E5=93=8D?= =?UTF-8?q?=E7=9C=8B=E6=9D=BF=E6=98=BE=E7=A4=BA=E6=B3=A8=E9=87=8A=E6=97=A0?= =?UTF-8?q?=E7=94=A8=E6=89=93=E5=8D=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../references/classification/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 11c3a04504..d484fde013 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') -- Gitee