fix: decrease warmup epoch to 1
Browse files
train.py
CHANGED
|
@@ -24,7 +24,7 @@ lambda_font = 2.0
|
|
| 24 |
lambda_direction = 0.5
|
| 25 |
lambda_regression = 1.0
|
| 26 |
|
| 27 |
-
num_warmup_epochs =
|
| 28 |
num_epochs = 100
|
| 29 |
|
| 30 |
log_every_n_steps = 100
|
|
|
|
| 24 |
lambda_direction = 0.5
|
| 25 |
lambda_regression = 1.0
|
| 26 |
|
| 27 |
+
num_warmup_epochs = 1
|
| 28 |
num_epochs = 100
|
| 29 |
|
| 30 |
log_every_n_steps = 100
|