SENet

SENet copied to clipboard

"What version of PyTorch are you using? My testing version may be different from the version of your pre-trained model, which could lead to code execution issues."

kya0a

https://github.com/sungonce/SENet/blob/7261e526c8b769296499bc2a620bd3cb0af57cf0/model/resnet.py#L191 class ResStage_basetransform(nn.Module): """Stage of ResNet.""" def __init__(self, w_in, w_out, stride, d, w_b=None, num_gs=1): super(ResStage_basetransform, self).__init__() #

hanklin3

What are the values of `dataset` and `gnd_fn` for a custom dataset? ``` if dataset == 'roxford5k': gnd_fn = 'gnd_roxford5k.pkl' elif dataset == 'rparis6k': gnd_fn = 'gnd_rparis6k.pkl' else: assert dataset...

harrytrinh96