folivo/thermal-conduction-ml
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix training submission script

Browse Source
This commit is contained in:
Filippo Olivo
2025-12-09 09:19:12 +01:00
parent f2ce282a68
commit c1820d5855
2 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
run.py
View File

@@ -5,7 +5,11 @@ torch.set_float32_matmul_precision("medium")
def main():
LightningCLI(subclass_mode_data=True, subclass_mode_model=True)
LightningCLI(
subclass_mode_data=True,
subclass_mode_model=True,
save_config_kwargs={"overwrite": True},
)
if __name__ == "__main__":

11
submit.sh
View File

@@ -1,8 +1,5 @@
#!/bin/bash
# python run.py fit --config experiments/config_4_layer_8_hidden.yaml
# python run.py fit --config experiments/config_8_layer_8_hidden.yaml
python run.py fit --config experiments/config_8_layer_16_hidden_refined.yaml
python run.py fit --config experiments/config_16_layer_8_hidden_refined.yaml
python run.py fit --config experiments/config_16_layer_16_hidden_refined.yaml
python run.py fit --config experiments/config_8_layer_16_hidden.yaml
# python run.py fit --config experiments/config_4_layer_16_hidden.yaml
export CUDA_VISIBLE_DEVICES=1
python run.py fit --config experiments/5_steps/config_16_layer_16_hidden_refined.yaml
python run.py fit --config experiments/5_steps/config_32_layer_16_hidden_refined.yaml
python run.py fit --config experiments/5_steps/config_8_layer_16_hidden_refined.yaml