Commit graph

24 commits

Author SHA1 Message Date
Corentin
b43b8b14d6 Trainer verbose argument implementation 2021-01-23 04:18:05 +09:00
Corentin
fa9188ad75 Fix dummy logger in trainer + stop mechanism 2021-01-20 13:16:01 +09:00
Corentin
9ac6fb64e8 Fix trainer validation loop 2021-01-13 00:14:04 +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
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
beeb4c1898 Add mypy_cache to .gitignore 2020-10-05 11:27:53 +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
95f2e52ff3 Fix SequenceGenerator for single sequence data 2020-07-06 13:20:31 +09:00
Corentin Risselin
58310ba077 HDF5 save and Conv2d stride typing change 2020-07-04 13:05:07 +09:00
Corentin Risselin
95bd1850b5 Change BatchGenerator save format from pickle to h5df 2020-04-28 16:54:44 +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
Corentin Risselin
9ab6adce7a Add MNIST loader 2020-03-06 14:06:40 +09:00
Corentin Risselin
846160e961 Initial commit 2020-03-06 14:05:09 +09:00