YanCote / IFT6268-simclr

Project for IFT6268
0 stars 0 forks source link

Yan/ft on xray #41

Closed YanCote closed 3 years ago

YanCote commented 3 years ago

Fix the W-CEL loss, now log HYPER in TB, new flag to use mlflow Assert on Label presence for each class, accuracy per class + avg, ad… Fix Assert on Data ratio, AOC works, before redesign of the loop Adding Loop Exec Time/CPU Mem usage Ready to start re-designing the loop Removed all Tensor in the eval code and the eval Fix another bug in WCEL, last eval removed Changes optimizer back to LARS Fix Printing + AOC again Added support to download MLFLow results Cleanup Finetuning Directory. First version of Distillation and infer… WIP: First version of inference which run with flower, ranamed config… Refactoring, saving the new layer in the module inference, Distillation run on flower, Next Steps: Validate XRAY pret…