Fork me on GitHub
摘要: VGG: 这个是VGG的网络模型架构: VGG( (features): Sequential( (0): Conv2d(3, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1)) (1): ReLU(inplace=True) (2): Co 阅读全文
posted @ 2021-07-27 14:09 sy- 阅读(833) 评论(2) 推荐(1) 编辑
摘要: Pytorch GPU运算过程中会出现:“cuda runtime error(2): out of memory”这样的错误。通常,这种错误是由于在循环中使用全局变量当做累加器,且累加梯度信息的缘故,用官方的说法就是:"accumulate history across your training 阅读全文
posted @ 2021-07-27 11:13 sy- 阅读(1455) 评论(0) 推荐(0) 编辑