GEM issue

[LightingMc]

No significant modifications to the code. Test run failed with the exps/gem.json provided.

File "/scratch/zar00002/PyCIL/main.py", line 31, in
main()
File "/scratch/zar00002/PyCIL/main.py", line 12, in main
train(args)
File "/scratch/zar00002/PyCIL/trainer.py", line 21, in train
_train(args)
File "/scratch/zar00002/PyCIL/trainer.py", line 93, in _train
model.incremental_train(data_manager)
File "/scratch/zar00002/PyCIL/models/gem.py", line 92, in incremental_train
self._train(self.train_loader, self.test_loader)
File "/scratch/zar00002/PyCIL/models/gem.py", line 123, in _train
self._update_representation(train_loader, test_loader, optimizer, scheduler)
File "/scratch/zar00002/PyCIL/models/gem.py", line 207, in update_representation
G[stpt:endpt, k].data.copy(params.grad.data.view(-1))
AttributeError: 'NoneType' object has no attribute 'data'