johnsk95 / PT4AL

Official PyTorch implementation of "PT4AL: Using Self-Supervised Pretext Tasks for Active Learning (ECCV2022)"
54 stars 6 forks source link

Question abou main.py testset #8

Closed lbg030 closed 1 year ago

lbg030 commented 1 year ago

Hi, thank you so much for your sharing your work.

main.py 44 line testset = Loader2(is_train=False, transform=transform_test)

I think it works trainset, so should I fix it like this? testset = Loader(is_train=False, transform=transform_test)

seominseok0429 commented 1 year ago

yes. i will fix it. (It was a long time ago, so I'll check it out and let you know the results.)