def extract_features(self, t_model, s_model, t_model_cfg, s_model_cfg):
    global t_FM
    t_FM = []
    global s_FM
    s_FM = []


    def hook1(module1, input, output):
        print("hooker Working")
        t_FM.append(output)

    handle = t_model.tbackbone.layer2[1].conv1.register_forward_hook(hook1)
    handle.remove()

    def hook2(module2, input, output):
        print("hooker Working")
        s_FM.append(output)

    handle = s_model.sbackbone.layer2[1].conv1.register_forward_hook(hook2)
    handle.remove()

    return t_FM, s_FM