文章目录

• 前言
• 代码实现
• • 1、导入数据库
  • • 关于torch.manual_seed(1)
  • 2、参数设置
  • 3、数据导入
  • 4、定义MMD损失
  • 5、定义训练函数
  • • 5.1 nn.CrossEntropyLoss()
    • 5.2 .detach()
    • 5.3 .size VS .shape
    • 5.4 .to(DEVICE)
    • 5.5 .max()
    • 5.6 optimizer.zero_grad()