torch.save(the_model.state_dict(), PATH) the_model ... TimeLogger import log import evaluate if t.cuda.is_available(): use_cuda = True else: ...
確定! 回上一頁