Commit ab7c13ca authored by Martin Řepa's avatar Martin Řepa

Learning epochs - change conf and add data_backup_setup file

parent 412f333e
......@@ -3,7 +3,7 @@ FROM ${IMAGE} as common
ENV APP_DIR /app
WORKDIR ${APP_DIR}
ARG DEVICE=gpu
ARG DEVICE=cuda
ENV device ${DEVICE}
# precompile wheels ----------------------------------------------------
......
experiments_per_setup: 6 # 6
legacy_folder: /home/ignac/experiments/learning_epochs/2000to30000by4000_8times # Change in regards with configuration
legacy_folder: /home/ignac/experiments/learning_epochs/2000to30000by4000_6times # Change in regards with configuration
epochs:
lower_bound: 2000
number_of_steps: 8 # 8 (6-8 should be enough)
......
......@@ -106,7 +106,14 @@ def exec_new_setup(conf: RootConfig, folder: str, iterations: int) -> Setup:
sub_results.append(sub_result)
i += 1
return Setup(cur_epochs, sub_results)
final_setup = Setup(cur_epochs, sub_results)
# ------------ Save backup data from this setup ---------
data_file = f'{folder}/backup_setup_data'
with open(data_file, 'wb') as file:
pickle.dump(final_setup, file)
# -------------------------------------------------------
return final_setup
def get_root_conf(conf_of_conf: dict) -> RootConfig:
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment