|
@@ -96,10 +96,10 @@ for i in range(config['training']['runs']):
|
|
|
|
|
|
# Save model
|
|
|
if not os.path.exists(
|
|
|
- config['paths']['model_output'] + '/' + str(config['model']['name'])
|
|
|
+ config['paths']['model_output'] + str(config['model']['name'] + '/models/')
|
|
|
):
|
|
|
os.makedirs(
|
|
|
- config['paths']['model_output'] + '/' + str(config['model']['name'])
|
|
|
+ config['paths']['model_output'] + str(config['model']['name']) + '/models/'
|
|
|
)
|
|
|
|
|
|
model_save_path = model_folder_path + 'models/' + str(i + 1) + '_s-' + str(seed)
|
|
@@ -114,5 +114,5 @@ for i in range(config['training']['runs']):
|
|
|
index=True,
|
|
|
)
|
|
|
|
|
|
- with open(model_save_path + 'summary.txt', 'a') as f:
|
|
|
+ with open(model_folder_path + 'summary.txt', 'a') as f:
|
|
|
f.write(f'{i + 1}: Test Accuracy: {tes_acc}\n')
|