diff --git a/requirements.txt b/requirements.txt index f258620..7e9d118 100644 --- a/requirements.txt +++ b/requirements.txt @@ -8,6 +8,7 @@ pandas wandb comet_ml hydra-core +safetensors black isort diff --git a/src/utils/init_utils.py b/src/utils/init_utils.py index 8c5aff7..aae2cec 100644 --- a/src/utils/init_utils.py +++ b/src/utils/init_utils.py @@ -142,7 +142,7 @@ def setup_saving_and_logging(config): """ Initialize the logger, writer, and saving directory. The saving directory is defined by the run_name and save_dir - arguments of config.writer and config.trainer, respectfully. + arguments of config.writer and config.trainer, respectively. Args: config (DictConfig): hydra config for the current experiment.