Add a Model - MichiganCOG/ViP GitHub Wiki

Add a Model

1. Create a new folder ViP/models/custom_model_name
2. Create a model class in ViP/models/custom_model_name/custom_model_name.py
   • Complete __init__, forward, and (optional) __load_pretrained_weights functions
3. Add PreprocessTrain and PreprocessEval classes within custom_model_name.py
4. Create config_train.yaml and config_test.yaml files for the new model