sungonce / senet Goto Github PK

Traceback (most recent call last):
  File "test.py", line 16, in <module>
    main()
  File "test.py", line 13, in main
    SENet_tester.__main__()
  File "/mount/sdc2_10TB/harry/tensorboard_embeddings/SENet/core/SENet_tester.py", line 30, in __main__
    checkpoint.load_checkpoint(cfg.TEST.WEIGHTS, model)
  File "/mount/sdc2_10TB/harry/tensorboard_embeddings/SENet/core/checkpoint.py", line 45, in load_checkpoint
    raise AssertionError("The model is not fully loaded.")
AssertionError: The model is not fully loaded.

def __init__(self, w_in, w_out, stride, d, w_b=None, num_gs=1):
    super(ResStage_basetransform, self).__init__()      # <------------------------
    for i in range(d):
        b_stride = stride if i == 0 else 1
        b_w_in = w_in if i == 0 else w_out
        trans_fun = get_trans_fun("basic_transform")      # <------------------------
        res_block = ResBlock(b_w_in, w_out, b_stride, trans_fun, w_b, num_gs)
        self.add_module("b{}".format(i + 1), res_block)

def forward(self, x):
    for block in self.children():
        x = block(x)
    return x

        if dataset == 'roxford5k':
            gnd_fn = 'gnd_roxford5k.pkl'
        elif dataset == 'rparis6k':
            gnd_fn = 'gnd_rparis6k.pkl'
        else:
            assert dataset