提交 3f071225 编写于 作者: Eric.Lee2021's avatar Eric.Lee2021 🚴🏻

Update train.py

上级 56ca4526
......@@ -64,7 +64,7 @@ def trainer(ops,f_log):
elif ops.model == "mobilenetv2":
model_ = MobileNetV2(num_classes=ops.num_classes , dropout_factor = ops.dropout)
elif ops.model == "ReXNetV1":
model_ = ReXNetV1(num_classes=ops.num_classes , width_mult=0.9, depth_mult=1.0, dropout_factor = ops.dropout)
model_ = ReXNetV1(num_classes=ops.num_classes , width_mult=1.0, depth_mult=1.0, dropout_factor = ops.dropout)
else:
print(" no support the model")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册