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

>>阅读原文<<