model = SomeModel() # 假設這是你按照Pytorch Lightning封裝好的Model. # 透過GPU訓練 trainer = Trainer(gpus=1, precision=16) trainer.fit(model) ...
確定! 回上一頁