mirror of
https://github.com/c-sooyoung/bo-ptycho.git
synced 2026-09-17 19:29:07 +09:00
fixes to results dir
This commit is contained in:
+1
-1
@@ -1,6 +1,6 @@
|
|||||||
io:
|
io:
|
||||||
input_data_path: '/home/swim/Si_project/data/Si2V1_1.mat'
|
input_data_path: '/home/swim/Si_project/data/Si2V1_1.mat'
|
||||||
result_dir: '/home/swim/Si_project/wrapper/results260721'
|
result_dir: '/home/swim/Si_project/wrapper/results/260721'
|
||||||
verbosity: 1
|
verbosity: 1
|
||||||
|
|
||||||
ptycho:
|
ptycho:
|
||||||
|
|||||||
@@ -1,15 +1,21 @@
|
|||||||
import os
|
import os
|
||||||
import sys
|
import sys
|
||||||
import yaml
|
import yaml
|
||||||
|
import shutil
|
||||||
import numpy as np
|
import numpy as np
|
||||||
|
|
||||||
import bo
|
import bo
|
||||||
import ptycho
|
import ptycho
|
||||||
|
|
||||||
def main(config):
|
def main(config_yaml):
|
||||||
|
|
||||||
|
with open(config_yaml, 'r') as f:
|
||||||
|
config = yaml.safe_load(f)
|
||||||
|
|
||||||
result_dir = config['io']['result_dir']
|
result_dir = config['io']['result_dir']
|
||||||
os.makedirs(result_dir, exist_ok=True)
|
os.makedirs(result_dir, exist_ok=True)
|
||||||
|
shutil.copy(config_yaml, os.path.join(result_dir, os.path.basename(config_yaml)))
|
||||||
|
|
||||||
|
|
||||||
randombo = bo.RandomBOEngine(config)
|
randombo = bo.RandomBOEngine(config)
|
||||||
|
|
||||||
@@ -49,7 +55,5 @@ if __name__ == "__main__":
|
|||||||
sys.exit(1)
|
sys.exit(1)
|
||||||
|
|
||||||
config_yaml = sys.argv[1]
|
config_yaml = sys.argv[1]
|
||||||
with open(config_yaml, 'r') as f:
|
|
||||||
config = yaml.safe_load(f)
|
|
||||||
|
|
||||||
main(config)
|
main(config_yaml)
|
||||||
|
|||||||
Reference in New Issue
Block a user