search

luo3300612 / image-captioning-DLCT

Official pytorch implementation of paper "Dual-Level Collaborative Transformer for Image Captioning" (AAAI 2021).
BSD 3-Clause "New" or "Revised" License
193 stars 31 forks source link

关于pth大小 #26

Closed haijie945 closed 2 years ago

haijie945 commented 2 years ago

请问跑完的模型是200M+嘛？我下载了您提供的pretrained_model.pth，240MB，为什么我跑出来有700M+。 size

luo3300612 commented 2 years ago

你好，这个是因为跑代码的时候除了保存模型还保存了优化器Adam的参数以便中断的时候继续训练，而我给的文件里只有模型本身的