Dassl.pytorch icon indicating copy to clipboard operation
Dassl.pytorch copied to clipboard

Add option to disable drop last on training

Open marco-garosi opened this issue 10 months ago • 0 comments

Hello, I appreciate your toolbox and I am actively using it! I have implemented a small change in the data loader construction in data_manager.py to force disabling drop_last on training datasets from the configuration file via the new DISABLE_DROP_LAST option. In some cases, one may not want to drop the last training batch, so this option allows the user to choose so. Note that the original behavior is left untouched by default, as this additional option has to be activated explicitly -- thus, it won't break existing code. I have implemented it for TRAIN_X, but it may be implemented for TRAIN_U as well.

marco-garosi avatar Jan 15 '25 08:01 marco-garosi