We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 2bdab29 commit 2c8d51cCopy full SHA for 2c8d51c
tutorials/train/segmentation/hrnet.py
@@ -39,7 +39,7 @@
39
# 浏览器打开 https://0.0.0.0:8001即可
40
# 其中0.0.0.0为本机访问,如为远程服务, 改成相应机器IP
41
num_classes = len(train_dataset.labels)
42
-model = pdx.seg.HRNet(num_classes=num_classes, width=64)
+model = pdx.seg.HRNet(num_classes=num_classes)
43
model.train(
44
num_epochs=20,
45
train_dataset=train_dataset,
0 commit comments