tdmpc2

tdmpc2 copied to clipboard

Hi, this line should be:

assert self.buffer.num_eps == (self.buffer.capacity // _cfg.episode_length), \
	f'Buffer has {self.buffer.num_eps} episodes, expected {self.buffer.capacity // _cfg.episode_length} episodes.'

Great catch! This seems to be a leftover from the previous buffer implementation (pre commit https://github.com/nicklashansen/tdmpc2/commit/54145a4d8c4c080836ff1f186fc5a87f70c8a8c7). I'll fix this soon and double-check that everything else in the offline trainer works as expected.

Closing this issue since it was fixed a while ago. Thanks again for reporting it!