corentin/torch_utils
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions
No description
28 commits 1 branch 0 tags 90 KiB
Find a file
Corentin 8ecf175265 Trainer source saving argument 
* Saving network state instead of whole instance
2021-02-13 03:02:03 +09:00
dataset Fix MNIST test data loading, add named buffer summary 2020-04-17 12:08:16 +09:00
utils Trainer implementation 2020-12-25 15:50:38 +09:00
.gitignore Add mypy_cache to .gitignore 2020-10-05 11:27:53 +09:00
layers.py Fix batch norm for conv2d 2021-01-30 02:11:06 +09:00
residual.py Residual blocks, precache for BatchGenerator 2020-09-09 02:27:59 +09:00
train.py Fix MNIST test data loading, add named buffer summary 2020-04-17 12:08:16 +09:00
trainer.py Trainer source saving argument 2021-02-13 03:02:03 +09:00