/DL_framework

Personal Pytorch framework to train and run deep learning algorithms

Primary LanguagePythonMIT LicenseMIT

DL_framework

Personal Pytorch framework to train and run deep learning algorithms

Installation

make install

Training

python framework/experiment.py train <path_to_config> <path_to_experiment_folder>

Testing

python framework/experiment.py test <path_to_config> <path_to_experiment_folder> <path_to_checkpoint_file>