Commit graph

11 commits

Author SHA1 Message Date
Corentin
f2282e3216 Trainer implementation
* Add Deconv2d
* Fix BatchGenerator save option when using current directory
2020-12-25 15:50:38 +09:00
corentin
4b786943f5 Fix batch norm training default value 2020-12-01 11:43:02 +09:00
Corentin
7f90010972 Add Conv3d layer 2020-11-26 18:50:38 +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 Risselin
7f4a162033 Fix Batch norm tracking 2020-07-07 11:04:57 +09:00
Corentin Risselin
58310ba077 HDF5 save and Conv2d stride typing change 2020-07-04 13:05:07 +09:00
Corentin Risselin
7db99ffa51 Fix MNIST test data loading, add named buffer summary 2020-04-17 12:08:16 +09:00
Corentin Risselin
ced13a4351 Clean mnist comment, add Linear layer 2020-04-17 10:45:13 +09:00
Corentin Risselin
12afc7cc93 Several improvement 2020-04-13 17:30:35 +09:00
Corentin Risselin
268429fa1a Layers, batch generator, memory 2020-03-31 13:46:01 +09:00