stupid_brain

前言:
本文用于记录本人AI新手期间犯的各种错误,时常更新。


正文开始:

  1. 读取数据的num_worker设置过少,以至于训练速度卡在读取数据上。
  2. 训练集数据处理:数据增强有利于解决过拟合问题。
  3. 模型:relu少写、batchnorm位置写错。
  4. test 记得关闭梯度更新with torch.no_grad():

你可能感兴趣的:(深度学习,深度学习,python,人工智能)