Closed wu-ming233 closed 9 months ago
@aik2mlj The new commit implements the changes requested above, as well as two minor things:
prepare_data.py
, and it's enabled by default.Let me know if anything else needs to be changed!
Looks good to me! Thank you for your efforts. Now it's only the augmentation range [-6, 7) that needs to be fixed.
This pull request entails:
collate_fn
shifting unevenly in dataloaderPolyffusionLearner
length=1
max_epochs
These changes are mainly just small additions to enhance user experience, and are completely backward-compatible.