如Tensorflow 一般,PyTorch 也擁有自己的distribute strategy 來做模型平行化和資料平行化的運算。 ... Linear(10, 10).to('cuda:0') self.relu = torch.nn.
確定! 回上一頁