-
Notifications
You must be signed in to change notification settings - Fork 31
Description
When trying this example code given for RegNet models:
#RegNetY600
model = get_model(model_name="RegNet",input_shape=input_shape,n_classes=n_classes,
verbose=True,fc_activation=fc_activation,stage_depth=[1,3,7,4],
stage_width=[48,112,256,608],stage_G=16,SEstyle_atten="SE")
This error shows up:
TypeError Traceback (most recent call last)
in ()
----> 1 build_model()1 frames
in build_model()
31 x = img_augmentation(inputs)
32 #RegNetY600
---> 33 model = get_model(model_name='regnety400')
34 # x = layers.GlobalAveragePooling2D()(x)
35 # x = layers.Dense(1)(x)/content/models/model_factory.py in get_model(model_name, input_shape, n_classes, verbose, dropout_rate, fc_activation, **kwargs)
93 n_classes=n_classes, dropout_rate=dropout_rate, fc_activation=fc_activation,
94 stage_depth=regnet_parameters[model_name['stage_depth']],stage_width=regnet_parameters[model_name['stage_width']],
---> 95 stage_G=regnet_parameters[model_name['stage_G']],SEstyle_atten=regnet_parameters[model_name['SEstyle_atten']],**kwargs).build()
96
97 if model_name in resnest3d_parameters.keys():TypeError: string indices must be integers