Issues
- 0
CLN, DOC, ENH: add complex examples
#28 opened by Xylambda - 0
ENH: use `enums` in conventions
#51 opened by Xylambda - 0
ENH: average parameters
#49 opened by Xylambda - 0
ENH: log with Accelerate
#45 opened by Xylambda - 0
ENH: divide callbacks module into submodules
#44 opened by Xylambda - 0
BUG: wrap schedulers in callbacks
#48 opened by Xylambda - 0
ENH: priority system for callbacks
#47 opened by Xylambda - 0
- 0
ENH: freeze and unfreeze model
#43 opened by Xylambda - 0
ENH, DISC: regularization as callbacks
#37 opened by Xylambda - 0
ENH: add predict method to the trainer
#38 opened by Xylambda - 0
- 0
- 0
ENH: use dunder methods
#16 opened by Xylambda - 0
DOC: fix doc
#20 opened by Xylambda - 0
TST, ENH: remake tests with single batch
#39 opened by Xylambda - 0
- 0
ENH: add logger and let user set level
#27 opened by Xylambda - 0
BLD: update README with pypi command
#32 opened by Xylambda - 0
DISC: integrate accelate
#26 opened by Xylambda - 0
ENH: gradient accumulation
#29 opened by Xylambda - 0
DISC: use rich instead of TQDM
#31 opened by Xylambda - 0
ENH: gradient clipping
#24 opened by Xylambda - 0
ENH: store results for all batches
#25 opened by Xylambda - 0
CLN: rename `features` for `batch`
#23 opened by Xylambda - 0
DOC: issues / typos
#17 opened by Xylambda - 0
TESTS: trainer and logger test fail
#19 opened by Xylambda - 0
ENH: support for mixed precision training
#21 opened by Xylambda - 1
- 0
- 0
TEST: update test
#15 opened by Xylambda - 0
- 0
ENH: learning rate finder
#13 opened by Xylambda - 0
ENH: add method to restart training params
#8 opened by Xylambda - 1
BUG: call of lr scheduler before optimizer
#10 opened by Xylambda - 1
ENH: compute loss for each batch
#4 opened by Xylambda - 0
ENH: support for early stopping
#1 opened by Xylambda - 0
ENH: support for regularization
#2 opened by Xylambda