Training the model use custom datasets

Question

Training the model use custom datasets

chensongloose opened this issue 2 years ago · 2 comments

Hello! Thanks for your great work! Excuse me, If I want to train custom datasets, how do I prepare the datasets and where do I modify the code?
Looking forward to your reply

Answer 1 · 2022-12-21T08:22:45.000Z

You need to write a custom pytorch dataset and update this line. If your image size is not 32x32, you also need to update model.py to fit your use case.

Answer 2 · 2023-03-15T05:33:03.000Z

Since there are no further questions, this issue is now closed.