torchvisoin.io read_image 和 读PIL或ndarry之后用transforms.ToTensor的区别

read_image:会正常读取图片
ToTensor：会将PIL或ndarray转成tensor，并且会将图像缩放到[0,1]之间

  1. 训练时，先用小数据集跑几个epoch，确保程序能够运行且不会出错
    1. 留意显存情况
    2. GPU使用率
    3. 温度
    4. evaluation时的显存情况
  2. model.eval()可能会引起显存增加，train时应该预留足够显存，避免evaluation时爆显存