【PyTorch】实现迁移学习框架DANN

文章目录

  • 前言
  • 代码实现
    • 1、导入数据库
      • 关于torch.manual_seed(1)
    • 2、参数设置
    • 3、数据导入
    • 4、定义训练函数
      • 4.1 nn.CrossEntropyLoss()
      • 4.2 .detach()
      • 4.3 .size VS .shape
      • 4.4 .to(DEVICE)
      • 4.5 .max()
      • 4.6 optimizer.zero_grad()
      • 4.7 len(data

你可能感兴趣的:(PyTorch,PHM项目实战--建模篇,pytorch,迁移学习,人工智能)