Skip to content
This repository was archived by the owner on May 22, 2024. It is now read-only.

Commit c977fb9

Browse files
authored
Update train.py
1 parent a5e835d commit c977fb9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

train_models/train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -132,7 +132,7 @@ def train(net_factory, prefix, end_epoch, base_dir,
132132
radio_cls_loss = 1.0;radio_bbox_loss = 0.5;radio_landmark_loss = 0.5;
133133
elif net == 'RNet':
134134
image_size = 24
135-
radio_cls_loss = 1.0;radio_bbox_loss = 0.5;radio_landmark_loss = 1.0;
135+
radio_cls_loss = 1.0;radio_bbox_loss = 0.5;radio_landmark_loss = 0.5;
136136
else:
137137
radio_cls_loss = 1.0;radio_bbox_loss = 0.5;radio_landmark_loss = 1.0;
138138
image_size = 48

0 commit comments

Comments
 (0)