Explorar o código

delete width=64

FlyingQianMM %!s(int64=5) %!d(string=hai) anos
pai
achega
2c8d51cbd4
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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,