commit
813b2b05af
|
@ -15,5 +15,5 @@ You will need to add files to `src/lib/datasets/sample/`, `src/lib/datasets/trai
|
||||||
|
|
||||||
## New architecture
|
## 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 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`.
|
- Add your model in `model_factory` of `src/lib/models/model.py`.
|
||||||
|
|
Loading…
Reference in New Issue