pytorch begin

Recommand approach for saving model

Stack overflow

• First

  1 2 3 4 5

  torch.save(model.state_dict(), PATH) # load model = Model(args) model.load_state_dict(torch.load(PATH))

• Second

  1 2 3 4

  torch.save(mode, PATH) # load model = torch.load(PATH)

Pytorch DataParallel

csdn