由於在訓練模型的時候,常常會使用GPU 來提升訓練速度,因此可藉由torch.device 設定分配的設備。 if torch.cuda.is_available(): cuda0 = torch.device(' ...
確定! 回上一頁