Skip to content

Add a Model

Jump to bottom
Eric Hofesmann edited this page Nov 15, 2019 · 2 revisions

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
Clone this wiki locally