Update models.py

This commit is contained in:
Glenn Jocher
2019-01-10 10:32:39 +01:00
committed by GitHub
parent 646a573740
commit 88804cad3b
+1 -1
View File
@@ -5,7 +5,7 @@ import torch.nn as nn
from utils.parse_config import * from utils.parse_config import *
from utils.utils import * from utils.utils import *
ONNX_EXPORT = True ONNX_EXPORT = False
def create_modules(module_defs): def create_modules(module_defs):