使用 :
為CPU中設置種子,生成隨機數:
torch.manual_seed(number)
為特定GPU設置種子,生成隨機數:
torch.cuda.manual_seed(number)
為所有GPU設置種子,生成隨機數:
torch.cuda.manual_seed_all(number)
使用原因 :
在需要生成隨機數據的實驗中,每次實驗都需要生成數據。設置隨機種子是為了確保每次生成固定的隨機數,這就使得每次實驗結果顯示一致了,有利於實驗的比較和改進。使得每次運行該 .py 文件時生成的隨機數相同。
