search

gjy3035 / C-3-Framework

An open-source PyTorch code for crowd counting
MIT License
698 stars 200 forks source link

About other pretrained models on datasets #60

Open ydlu opened 4 years ago

ydlu commented 4 years ago

你好，除了在GCC数据集上预训练好的模型，还会提供在其他数据集上预训练好的模型吗？比如在SHT_B上训练好的Res101模型？

另外，如果我想基于在SHT_B上训练好的Res101模型，继续在SHT_A数据集上进行训练，是不是设置config.py 中的 __C.PRE_GCC = False 参数为True及对应模型路径就可以了呢？和下面那个__C.RESUME 参数有点分不清。

gjy3035 commented 4 years ago

由于机器有限，暂时不会提供。

使用__C.PRE_GCC，只会加载模型参数。 使用 __C.RESUME，除了模型参数，训练阶段的任何数据都会加载。这个主要用于由于不可抗力原因导致模型中途训练终止的情况。