diff --git a/readme/DEVELOP.md b/readme/DEVELOP.md index 581b140..183a294 100644 --- a/readme/DEVELOP.md +++ b/readme/DEVELOP.md @@ -15,5 +15,5 @@ You will need to add files to `src/lib/datasets/sample/`, `src/lib/datasets/trai ## New architecture -- Add your model file to `src/lib/models/networks/`. The model should accept a dict `heads` of `{name: channels}`, which specify the name of each network output and its number of channals. Make sure your model returns a list (for multiple stages. Single stage model should return a list containing a single element.). The element of the list is a dict contraining the same keys with `heads`. -- Add your model in `model_factory` of `src/lib/models/model.py`. \ No newline at end of file +- Add your model file to `src/lib/models/networks/`. The model should accept a dict `heads` of `{name: channels}`, which specify the name of each network output and its number of channels. Make sure your model returns a list (for multiple stages. Single stage model should return a list containing a single element.). The element of the list is a dict contraining the same keys with `heads`. +- Add your model in `model_factory` of `src/lib/models/model.py`.