If None, run on device . gpus[0] should be the same gpu with device . Returns: A PyTorch model instance.
確定! 回上一頁