search

kekmodel / FixMatch-pytorch

Unofficial PyTorch implementation of "FixMatch: Simplifying Semi-Supervised Learning with Consistency and Confidence"
MIT License
758 stars 170 forks source link

custom dataset #17

Closed KirillHiddleston closed 4 years ago

KirillHiddleston commented 4 years ago

Do you have plans to make this repository compatible with a custom dataset, and if not, which files would need to be modified to do so?

kekmodel commented 4 years ago

You can modify cifar.py file for custom data.