Update config_pretraining.yaml
Browse files- config_pretraining.yaml +5 -6
config_pretraining.yaml
CHANGED
@@ -326,17 +326,16 @@ hardware:
|
|
326 |
output: ${oc.decode:${oc.env:OUTPUT_DIR}}
|
327 |
logs:
|
328 |
base: ${harware.paths.output}/logs
|
329 |
-
wandb: ${harware.paths.output}/logs
|
330 |
mlflow: ${harware.paths.output}/logs/mlflow
|
331 |
tensorboard: ${harware.paths.output}/logs/tensorboard
|
332 |
-
checkpoints: ${harware.paths.output}/checkpoint
|
333 |
-
plots: ${harware.paths.output}/plots
|
334 |
-
profiler: ${harware.paths.output}/profiler
|
335 |
-
graph: ${harware.paths.output}/graphs
|
336 |
files:
|
337 |
dataset: aifs-ea-an-oper-0001-mars-n320-1979-2022-6h-v6.zarr
|
338 |
dataset_land: aifs-ea-an-oper-0001-mars-n320-1979-2023-6h-v1-land.zarr
|
339 |
-
dataset_precip: aifs-od-an-oper-0001-mars-n320-2016-2023-6h-v2-precipitations.zarr
|
340 |
graph: graph_enc_proc_dec_n320.pt
|
341 |
checkpoint:
|
342 |
every_n_epochs: aifs-by_epoch-epoch_{epoch:03d}-val_wmse_{val_wmse:.3e}
|
|
|
326 |
output: ${oc.decode:${oc.env:OUTPUT_DIR}}
|
327 |
logs:
|
328 |
base: ${harware.paths.output}/logs
|
329 |
+
wandb: ${harware.paths.output}/logs/wandb
|
330 |
mlflow: ${harware.paths.output}/logs/mlflow
|
331 |
tensorboard: ${harware.paths.output}/logs/tensorboard
|
332 |
+
checkpoints: ${harware.paths.output}/checkpoint
|
333 |
+
plots: ${harware.paths.output}/plots
|
334 |
+
profiler: ${harware.paths.output}/profiler
|
335 |
+
graph: ${harware.paths.output}/graphs
|
336 |
files:
|
337 |
dataset: aifs-ea-an-oper-0001-mars-n320-1979-2022-6h-v6.zarr
|
338 |
dataset_land: aifs-ea-an-oper-0001-mars-n320-1979-2023-6h-v1-land.zarr
|
|
|
339 |
graph: graph_enc_proc_dec_n320.pt
|
340 |
checkpoint:
|
341 |
every_n_epochs: aifs-by_epoch-epoch_{epoch:03d}-val_wmse_{val_wmse:.3e}
|