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