Fix dataset size bug - Githubissues

sbdzdz / disco

Disentangled Continual Learning

https://arxiv.org/abs/2312.16731

Apache License 2.0

2 stars 1 forks source link

Fix dataset size bug #16

Closed sbdzdz closed 1 year ago

sbdzdz commented 1 year ago

Main changes:

fix an off-by-one error when setting dataset size
add support for int as shapes argument in the dataset constructor
add tests to catch this in the future