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