Commit graph

8 commits

Author SHA1 Message Date
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