Tags: Xylambda/torchfitter
Tags
ENH: regularization as callbacks (#42) * WIP: add callbacks scheme before and after loss step * DOC: add TODO * ENH: add 'predict' method to Trainer * BUG: fix minor error in predict * ENH, CLN: use black. Improve predict method. Fix bugs in preprocessing. * CLN, BUG: fix bug in LR sch callback. Fix flake8 complains. * CLN: use isort * CLN: minor code cleaning * BUG: fix bug on synchronization through cuda * DOC: update doc string for 'tabular_to_sliding_dataset' * ENH: allow multiple inputs for the model * ENH: add back '__predict_loader'. Remove some prediction callbacks * BLD: update NumPy requirements * CLN: remove Pandas * ENH: add missin __repr__. Add method to return history dict in trainer. Minor changes * ENH: add torhc distribution example * ENH, CLN: add loss update after loss step * ENH: store only the number of each batch loss * CLN: use isort * ENH: regularization as callbacks. Remove regularization module * BLD: update requirements to avoid conflicts * TST: update utils tests * CLN: update CHANGELOG Co-authored-by: Alejandro Pérez Sanjuán <aperez@ETS.ES>
ENH: regularization as callbacks (#42) * WIP: add callbacks scheme before and after loss step * DOC: add TODO * ENH: add 'predict' method to Trainer * BUG: fix minor error in predict * ENH, CLN: use black. Improve predict method. Fix bugs in preprocessing. * CLN, BUG: fix bug in LR sch callback. Fix flake8 complains. * CLN: use isort * CLN: minor code cleaning * BUG: fix bug on synchronization through cuda * DOC: update doc string for 'tabular_to_sliding_dataset' * ENH: allow multiple inputs for the model * ENH: add back '__predict_loader'. Remove some prediction callbacks * BLD: update NumPy requirements * CLN: remove Pandas * ENH: add missin __repr__. Add method to return history dict in trainer. Minor changes * ENH: add torhc distribution example * ENH, CLN: add loss update after loss step * ENH: store only the number of each batch loss * CLN: use isort * ENH: regularization as callbacks. Remove regularization module * BLD: update requirements to avoid conflicts * TST: update utils tests * CLN: update CHANGELOG Co-authored-by: Alejandro Pérez Sanjuán <aperez@ETS.ES>
WIP: bug fixes. Format code with Black (#33). Add StochasticWeightAve… …raging * WIP: bug fixes. Format code with Black * ENH: add SWA callback. Refactor utils.py module * BUG, DOC: minor fixes. Update Docstrings * ENH, TST: complete 'train_test_val_split'. Update setup * DOC: update README. Update CHANGELOG * BUG: fix SWA callback bug * ENH, BUG: fix log level bug. Add possibility to change logger name from child callback * CLN: update CHANGELOG. Fix docs
WIP, ENH: integrate accelerate. Update callback system (#30) * WIP, ENH: delegate device handling in accelerate * BLD, ENH: update requirements. Add gradient clipping * Add first version of gradient accumulation * ENH: use perf_counter. Refactor loss computation * Update setup. Add more hooks and methods to the callback system * BUG, ENH: fix minor bug on trainer. Update README. Add conventions * TST: update trainer tests * ENH: update MetricsHandler and InternalState to handle batch and epoch metrics * TST, DOC: update trainer docstring. Update trainer and callbacks tests * ENH: redesign history handling. Update tests * ENH: add Rich progress bar callback * ENH: redesign metrics handling * TST: fix minor bug in LearningRateScheduler * BLD, DOC: update requirements. Update docs
PreviousNext