pytorch之模型参数固定

训练集测试集划分固定 # 通过设置random_state设置固定比例 x_train, x_test, y_train, y_test = train_test_split(x_test, y, test_size = 0.1, random_state=seed) torch中的随机种子 # cpu torch.manual_seed(seed) # gpu torch.cuda.manual

>>阅读原文<<