CoTNet 修改笔记

CoTNet

依赖cupy

关于cupy的文章：

cupy_jacke121的专栏-CSDN博客_cupy

调用测试代码：

import torch

from models import cotnet50

if __name__ == '__main__':

    model = cotnet50(num_classes=100)

    state_dict = torch.load("cotnet50.pth.tar",map_location=torch.device('cpu'))

    new_state_dict = {}
    for k, v in state_dict.items():
        if 'module.' in k:
            new_state_dict[k[7:]] = v
        else:
            if "fc" 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(m