search

G-U-N / PyCIL

PyCIL: A Python Toolbox for Class-Incremental Learning
Other
815 stars 138 forks source link

pretrained model #86

Closed MoSheng37 closed 5 months ago

MoSheng37 commented 5 months ago

Can I incrementally train on top of the pre-trained ResNet weights for 7 classes? I haven't seen where in the code this is possible.

zhoudw-zdw commented 5 months ago

https://github.com/G-U-N/PyCIL/blob/a9b1ec8e3dbbe143cb72b008d7c8f946629d55b1/models/finetune.py#L39

You need to load model before incremental train. If you do not understand the code and do not want to modify it by yourself, maybe you can ask help from ChatGPT.