Pytorch自定義網絡結構+自定義數據加載+自定義Loss 全過程代碼示例

1、限定使用GPU的序號 import os os.environ['CUDA_VISIBLE_DEVICES'] = '3' os.system('echo $CUDA_VISIBLE_DEVICES') 2、導入相關頭文件 import torch import torch.nn as nn import torch.nn.functional as F import torch.utils.

>>阅读原文<<