美文网首页
tips总结

tips总结

作者: 幽并游侠儿_1425 | 来源:发表于2020-07-23 07:44 被阅读0次
    1. 如何print一个迭代器的所有值:
    dataset = list(dataset)
    print(dataset)
    
    1. transfer learning 需要改的内容
      (1) load trained weights
      (2) 在需要冷冻的层后加:
      for p in self.parameters():
      p.requires_grad = False
    2. GPU=0被占满,想用GPU=1来跑程序
      在最前面加 CUDA_VISIBLE_DEVICES=1
    3. 把整个文件夹从一个服务器转移到另一个服务器
      5.dataloader一直报错:
      Found 0 files in subfolders of: ./data/train Supported extensions are: .jpg,.jpeg,.png,.ppm,.bmp,.pgm,.tif,.tiff,.webp
      这时候如果数据在data/train里,写path的时候就写data就不会报错了

    相关文章

      网友评论

          本文标题:tips总结

          本文链接:https://www.haomeiwen.com/subject/puqclktx.html