search

DRSAD / iCaRL

96 stars 24 forks source link

保存模型有误 #5

Closed njulhy closed 3 years ago

njulhy commented 3 years ago

在iCaRL.py第187行应保存模型的OrderedDict参数，即使用torch.save(self.model.state_dict(), 'file_name') image