Commit History
Merge pull request #2 from shrits-ai/main
fb84786
atiwari751
commited on
Changed num_workers to 8 instead of 16
6e2d47c
Ubuntu
commited on
lower batch size and accumulation argument. Changed misclassified samples to be for last epoch only
2e9c13e
Ubuntu
commited on
Added augmentations as per the original paper
2959565
Ubuntu
commited on
Added graphs for top-1 and top-5 training and test accuracies. Added displaying of misclassified samples.
d759493
Ubuntu
commited on
added table to the training logs to copy to Excel
5005854
Ubuntu
commited on
Merge pull request #1 from shrits-ai/main
12b752d
atiwari751
commited on
Added checkpoint and early stopping
41b8141
Ubuntu
commited on
Modified code for imagent datase
d695662
Ubuntu
commited on