GwcNet/models/__init__.py
2019-04-14 17:34:58 +08:00

8 lines
152 B
Python

from models.gwcnet import GwcNet_G, GwcNet_GC
from models.loss import model_loss
__models__ = {
"gwcnet-g": GwcNet_G,
"gwcnet-gc": GwcNet_GC
}