fix issues with training in version 1 and sampling with custom dataset#194

Merged

WuJunde merged 2 commits intoSuperMedIntel:masterfrom pedropesserl:master

Oct 22, 2024

+30-10

Commits on Sep 27, 2024

copy load_part_state_dict from new to v1
pedropesserl
committed

Commits on Oct 22, 2024

fix sample with custom dataset
pedropesserl
committed