diff --git a/train.py b/train.py index 5277f3b939f1e9b07314e4fdb2b08ce19f09939d..1fee3d752f3ff283367cf9a0c30c9ec371cab1d8 100644 --- a/train.py +++ b/train.py @@ -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")