search

tornadomeet / ResNet

Reproduce ResNet-v2(Identity Mappings in Deep Residual Networks) with MXNet
Apache License 2.0
557 stars 199 forks source link

Update train_resnet.py #49

Open arundasan91 opened 6 years ago

arundasan91 commented 6 years ago

mx.model.FeedForward is deprecated in the latest master. Changing the same with mx.mod.Module. Kindly check and comment. Please feel free to make necessary changes.