Commit graph

53 commits

Author SHA1 Message Date
Corentin Risselin
6359258061 Small fix, making h5py optional 2023-01-04 16:58:48 +09:00
Corentin
1bac46219b Fix dropouts and typos in ViT 2021-08-17 15:54:35 +09:00
Corentin
0cf142571b Fix typos 2021-05-25 14:06:22 +09:00
Corentin
06db437aa4 Vision Transformer 2021-05-22 01:18:39 +09:00
Corentin
90abb84710 Merge branch 'hoel-bagard/torch_utils-BatchNormModifications' 2021-05-21 16:00:53 +09:00
Corentin
770a9a4f82 Avoid use_batch_norm as layers instance variable 2021-05-21 16:00:16 +09:00
Corentin
fe11f3e6d5 Merge branch 'master' into 'BatchNormModifications'
# Conflicts:
#   layers.py
2021-05-21 06:53:31 +00:00
Corentin
d87bb89e6c SSDLoss implementation 2021-05-21 15:14:14 +09:00
Corentin
092f4acc3b Add SSD 2021-05-17 21:05:15 +09:00
Corentin
8d13de5711 Improve ResNet layers 2021-04-29 19:45:32 +09:00
Corentin
9b0c8ec49d Fix IPCBatchGenerator SIGINT handling 2021-04-27 17:50:40 +09:00
Corentin
dcd99a55ea Add IPCBatchGenerator 2021-04-26 17:01:24 +09:00
Corentin
89240e417f Fix SequenceBatchGenerator's pipeline 2021-03-23 17:36:39 +09:00
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
d0fd6b2642 Index list parameter for SequenceBatchGenerator 2021-03-22 20:14:27 +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
51776d6999 Sequence stride option for SequenceBatchGenerator 2021-03-18 10:36:17 +09:00
Corentin
92971be5f0 Pre summary callback added to Trainer 2021-03-10 08:43:42 +09:00
Corentin
1fe6cd796e Fix last training loop 2021-03-01 23:56:15 +09:00
Corentin
50c395a07f Trainer last summary fix + memory utils 2021-02-25 02:18:02 +09:00
Corentin
24dea68044 Trainer save source parameter and batch generator fix 2021-02-19 13:35:51 +09:00
Corentin
8ecf175265 Trainer source saving argument
* Saving network state instead of whole instance
2021-02-13 03:02:03 +09:00
Corentin
42ae4474dd Fix batch norm for conv2d
* Change tensorboard folders
* Save root scripts in output code
2021-01-30 02:11:06 +09:00
Corentin
976f5cef49 Fix validation writer in Trainer
* Add last step summary (forced)
2021-01-27 02:04:12 +09:00
Corentin
d315f342a4 Accuracy function implemented in Trainer 2021-01-26 01:15:02 +09:00
Corentin
b43b8b14d6 Trainer verbose argument implementation 2021-01-23 04:18:05 +09:00
Hoel Bagard
ce6314bf5e
Fixed bias 2021-01-22 12:48:33 +09:00
Hoel Bagard
a4280a1b78
Fixed issues: layers now use self.use_batch_norm instead of default value, fixed Layer's forward 2021-01-22 12:38:07 +09:00
Hoel Bagard
54000b6c34 Fixed default use_batch_norm value 2021-01-21 20:36:22 +09:00
Hoel Bagard
7a6f5821bd Introduced the use_batch_norm variable, removed old code 2021-01-21 16:10:10 +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