Closed CaltropHungerton closed 1 year ago
Nice thanks! Adding a few comments
LGTM thanks!
Could you add a mention in the readme that the results are written to accuracy_results.csv
, speed_results.csv
and size_results.csv
?
LGTM thanks!
Could you add a mention in the readme that the results are written to
accuracy_results.csv
,speed_results.csv
andsize_results.csv
?
I also got rid of the TODO bullet point. I think there are a couple of other bullet points that we might be able to get rid of (add ohsumed dataset, has pytorch already been removed?) but you would know better than I would.
LGTM thanks! Could you add a mention in the readme that the results are written to
accuracy_results.csv
,speed_results.csv
andsize_results.csv
?I also got rid of the TODO bullet point. I think there are a couple of other bullet points that we might be able to get rid of (add ohsumed dataset, has pytorch already been removed?) but you would know better than I would.
Thanks! I'll clean up the todo.
added a function that exports the accuracy, speed, and size results as csv files.