Update export_model.py
parent
01f1314fbd
commit
4b10174a41
|
@ -38,7 +38,6 @@ class ExportModel(nn.Layer):
|
||||||
|
|
||||||
def __init__(self, config):
|
def __init__(self, config):
|
||||||
super().__init__()
|
super().__init__()
|
||||||
print (config)
|
|
||||||
self.base_model = build_model(config)
|
self.base_model = build_model(config)
|
||||||
|
|
||||||
# we should choose a final model to export
|
# we should choose a final model to export
|
||||||
|
|
Loading…
Reference in New Issue