如果你的模型架構保持不變、輸入大小保持不變,設置torch.backends.cudnn.benchmark = True。 7. 小心CPU 和GPU 之間頻繁的數據傳輸. 當頻繁地使用tensor.
確定! 回上一頁