Commit graph

14 commits

Author SHA1 Message Date
Corentin
73457750a8 Add pipeline to SequenceBatchGenerator
* Remove flip_data parameter (can be done in the pipeline)
2021-03-23 02:13:40 +09:00
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
Corentin
24dea68044 Trainer save source parameter and batch generator fix 2021-02-19 13:35:51 +09:00
Corentin
f2282e3216 Trainer implementation
* Add Deconv2d
* Fix BatchGenerator save option when using current directory
2020-12-25 15:50:38 +09:00
corentin
144ff4a004 Data flip implementation 2020-12-18 20:30:36 +09:00
Corentin
8a0d9b51a3 Improve code 2020-12-18 17:40:39 +09:00
Corentin
128dfe511e Fix batch generator when data/label isn't ndarray 2020-12-18 16:24:21 +09:00
Corentin
379dd4814f Fix batch generators last batch size 2020-12-04 11:28:12 +09:00
Corentin
40ebd0010d Implement num_workers for batch generators
* Clean and factor code in batch generators
2020-12-04 04:14:13 +09:00
Corentin
5dee78b729 Add Conv1d layer, update SequenceBatchGenerator 2020-11-26 17:05:19 +09:00
Corentin Risselin
5081cf63fe Residual blocks, precache for BatchGenerator 2020-09-09 02:27:59 +09:00
corentin
95f2e52ff3 Fix SequenceGenerator for single sequence data 2020-07-06 13:20:31 +09:00
Corentin Risselin
95bd1850b5 Change BatchGenerator save format from pickle to h5df 2020-04-28 16:54:44 +09:00
Corentin Risselin
268429fa1a Layers, batch generator, memory 2020-03-31 13:46:01 +09:00