torch_utils/utils
Corentin 86787f6517 Pipeline implementation for BatchGenerator
* Add end_epoch_callback in Trainer
* Fix Layer.ACTIVATION in cas of nn.Module
2021-03-18 21:30:59 +09:00
..
batch_generator.py Pipeline implementation for BatchGenerator 2021-03-18 21:30:59 +09:00
colored_formatter.py Initial commit 2020-03-06 14:05:09 +09:00
console_color.py Initial commit 2020-03-06 14:05:09 +09:00
logger.py Initial commit 2020-03-06 14:05:09 +09:00
memory.py Trainer last summary fix + memory utils 2021-02-25 02:18:02 +09:00
sequence_batch_generator.py Sequence stride option for SequenceBatchGenerator 2021-03-18 10:36:17 +09:00