Commit 12af179a authored by Martino Bertoni's avatar Martino Bertoni 🌋
Browse files

fixed GLOBAL model loading

parent 09ef7e48
......@@ -59,7 +59,12 @@ class Signaturizer(object):
else:
self.model_names = [model_name]
else:
self.model_names = model_name
if model_name == ['GLOBAL']:
self.model_names = [y + x for y in 'ABCDE' for x in '12345']
else:
if 'GLOBAL' in model_name:
raise Exception('"GLOBAL" model can only be used alone.')
self.model_names = model_name
# load modules as layer to compose a new model
main_input = Input(shape=(2048,), dtype=tf.float32, name='main_input')
sign_output = list()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment