- 如何print一个迭代器的所有值：
dataset = list(dataset)
print(dataset)
- transfer learning 需要改的内容
(1) load trained weights
(2) 在需要冷冻的层后加：
for p in self.parameters():
p.requires_grad = False - GPU=0被占满，想用GPU=1来跑程序
在最前面加CUDA_VISIBLE_DEVICES=1 - 把整个文件夹从一个服务器转移到另一个服务器
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就不会报错了
网友评论