Skip to content

fix: use dynamic num_classes for random labels in training#45

Open
iago-suarez wants to merge 1 commit intowillisma:mainfrom
iago-suarez:main
Open

fix: use dynamic num_classes for random labels in training#45
iago-suarez wants to merge 1 commit intowillisma:mainfrom
iago-suarez:main

Conversation

@iago-suarez
Copy link

Dear @willisma,

Thanks for sharing your awesome work. It has been a pleasure to read through it. I noticed a small inconsistency in the training script. Although the number of classes can be configured, the default value (1000) is still hardcoded in a few other places. This causes an error when the number of classes is set to a value smaller than 1000.

It might be safer to rely on the dynamically defined number of classes throughout the script to avoid this issue.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant