fix: adjust back lr and increase warmup a bit
Browse files
train.py
CHANGED
|
@@ -22,18 +22,18 @@ final_batch_size = 128
|
|
| 22 |
single_device_num_workers = 24
|
| 23 |
|
| 24 |
|
| 25 |
-
lr = 0.
|
| 26 |
b1 = 0.9
|
| 27 |
b2 = 0.999
|
| 28 |
|
| 29 |
-
lambda_font =
|
| 30 |
lambda_direction = 0.5
|
| 31 |
lambda_regression = 1.0
|
| 32 |
|
| 33 |
regression_use_tanh = True
|
| 34 |
augmentation = True
|
| 35 |
|
| 36 |
-
num_warmup_epochs =
|
| 37 |
num_epochs = 100
|
| 38 |
|
| 39 |
log_every_n_steps = 100
|
|
|
|
| 22 |
single_device_num_workers = 24
|
| 23 |
|
| 24 |
|
| 25 |
+
lr = 0.0001
|
| 26 |
b1 = 0.9
|
| 27 |
b2 = 0.999
|
| 28 |
|
| 29 |
+
lambda_font = 2.0
|
| 30 |
lambda_direction = 0.5
|
| 31 |
lambda_regression = 1.0
|
| 32 |
|
| 33 |
regression_use_tanh = True
|
| 34 |
augmentation = True
|
| 35 |
|
| 36 |
+
num_warmup_epochs = 5
|
| 37 |
num_epochs = 100
|
| 38 |
|
| 39 |
log_every_n_steps = 100
|