Convolutional-MLPs改造

Convolutional-MLPs 

人脸识别用，112*112返回7*7特征：

调用代码：

import torch

from src import convmlp_s

if __name__ == '__main__':
    model = convmlp_s(num_classes=100)

    state_dict = torch.load("convmlp_s_imagenet.pth")

    new_state_dict = {}
    for k, v in state_dict.items():
        if 'module.' in k:
            new_state_dict[k[7:]] = v
        else:
            if "head" in k:
                continue
            new_state_dict[k] = v

    model.load_state_dict(new_state_dict, strict=False)


    # model.load_state_dict(state_dict,strict=False)

    model.eval()
    model_path = "dicenet.pth"
    torch.save(model.state_dict(), model_path)

    import os
    import time

    fsize = os.path.getsize(model_path)
    fsize = fsize / float(1024 * 1024)

    print(f"model size {round(fs