def get_record_keeper(csv_folder, tensorboard_folder=None, global_db_path=None, experiment_name=None, is_new_experiment=True, save_figures=False, save_lists=False): try: import record_keeper as record_keeper_package from torch.utils.tensorboard import SummaryWriter record_writer = record_keeper_package.RecordWriter( folder=csv_folder, global_db_path=global_db_path, experiment_name=experiment_name, is_new_experiment=is_new_experiment, save_lists=save_lists) tensorboard_writer = SummaryWriter( log_dir=tensorboard_folder ) if tensorboard_folder is not None else None record_keeper = record_keeper_package.RecordKeeper( tensorboard_writer=tensorboard_writer, record_writer=record_writer, attributes_to_search_for=c_f. list_of_recordable_attributes_list_names(), save_figures=save_figures) return record_keeper, record_writer, tensorboard_writer except ModuleNotFoundError as e: logging.warn(e) logging.warn("There won't be any logging or model saving.") logging.warn("To fix this, pip install record-keeper tensorboard") return None, None, None
def get_record_keeper(csv_folder, tensorboard_folder, global_db_path=None, experiment_name=None, is_new_experiment=True): try: import record_keeper as record_keeper_package from torch.utils.tensorboard import SummaryWriter record_writer = record_keeper_package.RecordWriter(csv_folder, global_db_path, experiment_name, is_new_experiment) tensorboard_writer = SummaryWriter(log_dir=tensorboard_folder) record_keeper = record_keeper_package.RecordKeeper(tensorboard_writer, record_writer, ["record_these", "learnable_param_names"]) return record_keeper, record_writer, tensorboard_writer except ModuleNotFoundError as e: logging.warn(e) return None, None, None
def get_record_keeper(csv_folder, tensorboard_folder, global_db_path=None, experiment_name=None, is_new_experiment=True, save_figures=False): try: import record_keeper as record_keeper_package from torch.utils.tensorboard import SummaryWriter record_writer = record_keeper_package.RecordWriter( csv_folder, global_db_path, experiment_name, is_new_experiment) tensorboard_writer = SummaryWriter(log_dir=tensorboard_folder) record_keeper = record_keeper_package.RecordKeeper( tensorboard_writer, record_writer, ["record_these", "learnable_param_names"], save_figures=save_figures) return record_keeper, record_writer, tensorboard_writer except ModuleNotFoundError as e: logging.warn(e) logging.warn("There won't be any logging or model saving.") logging.warn("To fix this, pip install record-keeper tensorboard") return None, None, None