GwcNet/models/__init__.py

8 lines
152 B
Python
Raw Normal View History

2019-04-14 17:34:58 +08:00
from models.gwcnet import GwcNet_G, GwcNet_GC
from models.loss import model_loss
__models__ = {
"gwcnet-g": GwcNet_G,
"gwcnet-gc": GwcNet_GC
}