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.

>>阅读原文<<