使用resnet50训练图像分类,保存的模型大小为220M左右,正常resnet50模型大小不是90多么,为什么会这么大,可以做一下轻量化吗
我们默认会把训练参数的梯度存下来,所以基本是二倍的关系,可以通过ckpt config 配置只保存推理网参数
使用resnet50训练图像分类,保存的模型大小为220M左右,正常resnet50模型大小不是90多么,为什么会这么大,可以做一下轻量化吗
我们默认会把训练参数的梯度存下来,所以基本是二倍的关系,可以通过ckpt config 配置只保存推理网参数