I use a pretraned network from github.
net = WSDAN(num_classes=2, M=8, net="efficientnet-b3", pretrained=False)
ckpt = torch.load("./pretrained/abcd.pth", map_location="cpu")
net.load_state_dict(ckpt["state_dict"])