search

microsoft / only_train_once

OTOv1-v3, NeurIPS, ICLR, TMLR, DNN Training, Compression, Structured Pruning, Erasing Operators, CNN, Diffusion, LLM
MIT License
24 stars 5 forks source link

chore: update metric logging #11

Closed aponte411 closed 2 months ago

aponte411 commented 2 months ago

Instead of manually calculating and logging optimization metrics, use the new compute_metrics function from HESSO