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
|
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
|
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 |
|
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 |
|