Ver Fonte

delete width=64

FlyingQianMM há 5 anos atrás
pai
commit
2c8d51cbd4
1 ficheiros alterados com 1 adições e 1 exclusões
  1. 1 1
      tutorials/train/segmentation/hrnet.py

+ 1 - 1
tutorials/train/segmentation/hrnet.py

@@ -39,7 +39,7 @@ eval_dataset = pdx.datasets.SegDataset(
 # 浏览器打开 https://0.0.0.0:8001即可
 # 其中0.0.0.0为本机访问,如为远程服务, 改成相应机器IP
 num_classes = len(train_dataset.labels)
-model = pdx.seg.HRNet(num_classes=num_classes, width=64)
+model = pdx.seg.HRNet(num_classes=num_classes)
 model.train(
     num_epochs=20,
     train_dataset=train_dataset,