-
Notifications
You must be signed in to change notification settings - Fork 298
Design new model
Tang, Wenyi edited this page May 10, 2019
·
1 revision
version 0.0.1-alpha
- Create a new python file in VSR.Models.
- Copy codes from
Edsr.py
and rename the classEDSR
. - Write model graph in function
build_graph
- Write loss in function
build_loss
(You can also write inbuild_graph
and ignorebuild_loss
, that's your choice.) - (Optional) Write summaries and savers. (There're default summary and saver)
- Open
VSR.Models.__init__.py
. - Add an entry to
models
dict. The entry isalias: (file-name, class-name)
.
- Create a new python file in VSRTorch.Models.
- Copy codes from
Espcn.py
and rename the classESPCN
. - Write modules.
- Write forward and backward data-path in function
train
. - Write forward data-path in function
eval
.
- Open
VSRTorch.Models.__init__.py
. - Add an entry to
models
dict. The entry isalias: (file-name, class-name)
.