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

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