Ashaar / poetry_diacritizer /create_configs.py
Ababababababbababa's picture
Duplicate from arbml/Ashaar
6faf7e7
raw
history blame
No virus
401 Bytes
import yaml
fname = "config/gpt-cls-tash-proc.yml"
stream = open(fname, 'r')
data = yaml.load(stream, Loader=yaml.FullLoader)
for i in range(0, 10):
data['n_layer'] = i
data['log_directory'] = f'log_dir_cls_{i}_tash_proc'
data['max_steps'] = 5000
with open(f"config/gpt-cls-{i}-tash-proc.yml", 'w') as yaml_file:
yaml_file.write( yaml.dump(data, default_flow_style=False))