search

broadinstitute / keras-resnet

Keras package for deep residual networks
Other
300 stars 127 forks source link

The freeze_bn parameter does not work for TimeDistributed models #32

Open jccaicedo opened 6 years ago

jccaicedo commented 6 years ago

I get the following error:

    net = keras_resnet.models.TimeDistributedResNet18(input_set, include_top=False, freeze_bn=True)
  File "/home/ubuntu/.pyenv/versions/3.6.2/lib/python3.6/site-packages/keras_resnet/models/_time_distributed_2d.py", line 125, in TimeDistributedResNet18
    return TimeDistributedResNet(inputs, blocks, block=keras_resnet.blocks.time_distributed_basic_2d, include_top=include_top, classes=classes, *args, **kwargs)
  File "/home/ubuntu/.pyenv/versions/3.6.2/lib/python3.6/site-packages/keras_resnet/models/_time_distributed_2d.py", line 87, in TimeDistributedResNet
    return keras.models.Model(inputs=inputs, outputs=outputs, *args, **kwargs)
  File "/home/ubuntu/.pyenv/versions/3.6.2/lib/python3.6/site-packages/keras/legacy/interfaces.py", line 87, in wrapper
    return func(*args, **kwargs)
TypeError: __init__() got an unexpected keyword argument 'freeze_bn'

Also, the default value for freeze_bn is different in for 2D and TimeDistributed models.