File size: 401 Bytes
6faf7e7
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
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))