Skip to content

Commit f59a293

Browse files
committed
export pc_aug parameter
1 parent 885082f commit f59a293

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

export_ckpt.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,8 +47,13 @@ def export(exp_name, ckpt_path, save_path, work_dir):
4747
new_config['win_size'] = config['win_size']
4848
new_config['fmin'] = config['fmin']
4949
new_config['fmax'] = config['fmax']
50+
if 'pc_aug' not in config.keys():
51+
new_config['pc_aug'] = False
52+
else:
53+
new_config['pc_aug'] = config['pc_aug']
5054
if 'mini_nsf' not in new_config.keys():
5155
new_config['mini_nsf'] = False
56+
5257
json_file.write(json.dumps(new_config, indent=1))
5358
print("Export configuration file successfully: ", new_config_file)
5459

0 commit comments

Comments
 (0)