msamogh/nonechucks

Override some functions of the original dataset

yanxu1per opened this issue · 2 comments

Hi, thank you for your work.
I am using nonechucks to deal with bad examples. However, in my original dataset class, one function get_img_info is defined. After using nc.SafeDataset to wrap it, it shows, 'SafeDataset object has no attribute 'get_img_info'

Before that, I change _DataLoaderIter to _BaseDataLoaderIter in your code because the former one doesn't exist in my torch version (1.1.0). Weird

Thanks for raising the issue. It's been fixed now. 26382f0