First of all thanks for the help.
I changed my code like this:
model = torch.nn.DataParallel(resnet.__dict__[args.arch]())

    for param in model.parameters():
        param.requires_grad = False
    model.fc = nn.Linear(64, 10)

    model.cuda()

Then I load the checkpoint but it gives this error:
RuntimeError: Error(s) in loading state_dict for DataParallel:
Missing key(s) in state_dict: “fc.weight”, “fc.bias”.
Excuse me, I’m pretty new in pytorch.