yingkaisha commited on
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 70
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 70
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}