Corentin
|
06db437aa4
|
Vision Transformer
|
2021-05-22 01:18:39 +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
|
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
|
42ae4474dd
|
Fix batch norm for conv2d
* Change tensorboard folders
* Save root scripts in output code
|
2021-01-30 02:11:06 +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
|
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 |
|