torch.manual_seed()

  torch.manual_seed(args.seed)   # 爲CPU設置種子用於生成隨機數,以使得結果是確定的。 當你設置一個隨機種子時,接下來的隨機算法生成數根據當前的隨機種子按照一定規律生成。 隨機種子作用域是在設置時到下一次設置時。要想重複實驗結果,設置同樣隨機種子即可。   修改隨機種子數,可以看到發生了變化。而且即使關閉進程,重新運行代碼,同一隨機種子下生成的數依舊不變:
本站公眾號
   歡迎關注本站公眾號,獲取更多信息