Commit ·
4ba0e92
1
Parent(s): 3f90e98
update config
Browse files
FuXi-sigma-base/configs/model_single.yml
CHANGED
|
@@ -105,7 +105,7 @@ trainer:
|
|
| 105 |
start_epoch: 0
|
| 106 |
|
| 107 |
reload_epoch: True
|
| 108 |
-
epochs: &epochs
|
| 109 |
|
| 110 |
use_scheduler: True
|
| 111 |
scheduler: {'scheduler_type': 'cosine-annealing', 'T_max': *epochs, 'last_epoch': -1}
|
|
|
|
| 105 |
start_epoch: 0
|
| 106 |
|
| 107 |
reload_epoch: True
|
| 108 |
+
epochs: &epochs 170
|
| 109 |
|
| 110 |
use_scheduler: True
|
| 111 |
scheduler: {'scheduler_type': 'cosine-annealing', 'T_max': *epochs, 'last_epoch': -1}
|
FuXi-sigma-physics/configs/model_single.yml
CHANGED
|
@@ -105,7 +105,7 @@ trainer:
|
|
| 105 |
start_epoch: 0
|
| 106 |
|
| 107 |
reload_epoch: True
|
| 108 |
-
epochs: &epochs
|
| 109 |
|
| 110 |
use_scheduler: True
|
| 111 |
scheduler: {'scheduler_type': 'cosine-annealing', 'T_max': *epochs, 'last_epoch': -1}
|
|
|
|
| 105 |
start_epoch: 0
|
| 106 |
|
| 107 |
reload_epoch: True
|
| 108 |
+
epochs: &epochs 170
|
| 109 |
|
| 110 |
use_scheduler: True
|
| 111 |
scheduler: {'scheduler_type': 'cosine-annealing', 'T_max': *epochs, 'last_epoch': -1}
|