donlee90 / icarl

PyTorch Implementation of iCaRL
103 stars 34 forks source link

the output dim of network and the issue of loss #9

Open SongYang008 opened 9 months ago

SongYang008 commented 9 months ago

hi, I read your code and I have two questions the first one: in this paper, the author didn't mentioned to add the dim of output when training on new tasks, but your code added nodes for fully conneted layer when training on new tasks. the second one: in this paper, classification loss has two parts, one is about traditional CE Loss, and another is the CE Loss with none target labels and "1 - prediction". Thanks for your attention!