Releases: xiahouzuoxin/torchctr
Releases · xiahouzuoxin/torchctr
Move to polars, improve Trainer
Full Changelog: v0.1.3...v0.2.0
- FeatureTransform:
polarsis preferred over pandas- add feature cross, feature pre-process by polars' Expr
- fix lots of bugs
- Trainer:
- add
accelerateto support the trainer on GPU - add
tensorboardsupport - save only the model state dict instead of the pickle file to help loading the model on another device
- add
v0.1.3
The last version that feature transform by pandas.
Full Changelog: v0.1.2...v0.1.3
v0.1.2
- Fixed the OOM problem of DataFrameDataset by replacing list with numpy, and process all data to numpy before
__get_item__ - Default value changes of
DataFrameDataset:is_rawparameter default to Truelist_padding_valueparameter default to None. Therefore, you need to set thelist_padding_valuewhen init DataFrameDataset or usingcollate_fnin DataFrameDataset when data loading if you have var-length sequence features
- Support string input type of sequence features (comma split)
Full Changelog: v0.1.1...v0.1.2
v0.1.1
Full Changelog: v0.1.0...v0.1.1
Fixed a minor problem when input sequence features are the same length.
v0.1.0
Full Changelog: https://github.com/xiahouzuoxin/torchrec/commits/v0.1.0
The first release with all features implemented as README.md.