search

xvjiarui / GCNet

GCNet: Non-local Networks Meet Squeeze-Excitation Networks and Beyond
Apache License 2.0
1.19k stars 165 forks source link

How can I use gc block in resnet18? #34

Closed LJXLJXLJX closed 4 years ago

LJXLJXLJX commented 4 years ago

Considering the limitations of hardware performance, I tried to train a lightweight model. So I use resnet18. But resnet18 use basicblock instead of bottleneck, but it seems that you don't inplement it for basicblock. If I want to use gc block in basicblock , how can I designed it? Thank you !

xvjiarui commented 4 years ago

Hi @LJXLJXLJX , Sorry for the late reply. I think add GC block just before the skip connection will be fine.

LJXLJXLJX commented 4 years ago

@xvjiarui Thank you, it worked, but took a really long time to converge.

atnegam commented 2 years ago

@LJXLJXLJX Hello, what is your initial learning rate?