search

arthurdouillard / incremental_learning.pytorch

A collection of incremental learning paper implementations including PODNet (ECCV20) and Ghost (CVPR-W21).
MIT License
383 stars 60 forks source link

Python and PyTorch version #29

Closed luhaochun closed 3 years ago

luhaochun commented 3 years ago

Hi, I was wondering if you could provide the Python and PyTorch versions that this code was tested with?

arthurdouillard commented 3 years ago

Python is 3.6, Pytorch is 1.2.0.

Please look at the requirements: https://github.com/arthurdouillard/incremental_learning.pytorch/blob/master/requirements.txt#L1

Anyway, the code should work for later version of Pytorch.

arthurdouillard commented 3 years ago

If testing PODNet NME (PODNet CNN should be fine) or BiC, try first with torch version 1.2.0 (https://github.com/arthurdouillard/incremental_learning.pytorch/issues/31). I'm not sure yet why.