fix: adjust strategy: data aug+no tanh
Browse files
train.py
CHANGED
|
@@ -30,7 +30,7 @@ lambda_font = 2.0
|
|
| 30 |
lambda_direction = 0.5
|
| 31 |
lambda_regression = 1.0
|
| 32 |
|
| 33 |
-
regression_use_tanh =
|
| 34 |
augmentation = True
|
| 35 |
|
| 36 |
num_warmup_epochs = 5
|
|
|
|
| 30 |
lambda_direction = 0.5
|
| 31 |
lambda_regression = 1.0
|
| 32 |
|
| 33 |
+
regression_use_tanh = False
|
| 34 |
augmentation = True
|
| 35 |
|
| 36 |
num_warmup_epochs = 5
|