Spaces:
Sleeping
Sleeping
File size: 1,003 Bytes
9c4b1c4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
from parser import get_parser
import subprocess
if __name__ == "__main__":
parser = get_parser()
settings = parser.parse_args()
print(settings)
with open ("configs/train_template.json", "r") as f:
training_template = f.read()
training_template = training_template.replace("${DATA_KEYS}", settings.data_keys)
training_template = training_template.replace("${DATA_ROOT}", settings.data_root)
training_template = training_template.replace("${SPLIT_FILE}", settings.split_file)
training_template = training_template.replace("${NAME}", settings.name)
training_template = training_template.replace("${DEVICE}", settings.device)
training_template = training_template.replace("${TASK}", settings.task)
with open("configs/train.json", "w") as f:
f.write(training_template)
print("Train config file created")
subprocess.run(f'python -u src/train.py --config configs/train.json', shell=True)
|