Skip to content

Commit 2c8d51c

Browse files
committed
delete width=64
1 parent 2bdab29 commit 2c8d51c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tutorials/train/segmentation/hrnet.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@
3939
# 浏览器打开 https://0.0.0.0:8001即可
4040
# 其中0.0.0.0为本机访问,如为远程服务, 改成相应机器IP
4141
num_classes = len(train_dataset.labels)
42-
model = pdx.seg.HRNet(num_classes=num_classes, width=64)
42+
model = pdx.seg.HRNet(num_classes=num_classes)
4343
model.train(
4444
num_epochs=20,
4545
train_dataset=train_dataset,

0 commit comments

Comments
 (0)