Releases: analokmaus/kuma_utils
Releases · analokmaus/kuma_utils
v0.7.0
v0.7.0b
update settings related to rye
v0.6.2
What's new
- Updated pytorch dependencies for security reasons
v0.6.1
What's new
- Updated dependencies
v0.6.0
What's New
- TorchTrainer can now accept models in which loss functions are included (e.g., ArcFace, LLMs). You must pass
criterion = None
to theTorchTrainer.train()
and calculate the loss inside your training hook.
Sample code
def forward_train(self, trainer, inputs):
embeddings, loss = trainer.model.*inputs)
return loss, embeddings.detach()
v0.5.1
v0.5.0
v0.4.2
What's new
- Wandb run name will be set to TorchTrainer.serial when it is not specified.
v0.4.1
What's new
- Bugfixes for TorchTrainer
- Wandb run name will be
TorchTrainer.serial
by default. Specify name inwandb_params
to override.
v0.4.0
Update
LGBMImputer
- Cross validation support (see tutorial notebook)
- New argument
target_cols
, in case you want to fit columns with no missing values. - Internal label encoder is added. Now you don't have to encode categorical variables in advance.
TemperatureScaler
- Moved to
kuma_utils.torch.modules.calibration
Others
- Bugfixes