Releases: HazyResearch/metal
Releases · HazyResearch/metal
v0.3.1
v0.3.0
- Adds logging via a
LogWriter
class (and in contrib, aTensorBoardWriter
class) - Refactors the
ModelTuner
class - Renames
Classifier.train
toClassifier.train_model
to avoid overwriting thenn.Module.train
function - Adds support for metrics requiring predicted probabilities (like ROC-AUC)
- Better handling of
EndModel
construction - Other fixes and additions focused on
EndModel
training