corentin/torch_utils
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions
No description
49 commits 1 branch 0 tags 90 KiB
  • Python 100%
Find a file
Corentin 90abb84710 Merge branch 'hoel-bagard/torch_utils-BatchNormModifications'
2021-05-21 16:00:53 +09:00
dataset Fix MNIST test data loading, add named buffer summary 2020-04-17 12:08:16 +09:00
ssd SSDLoss implementation 2021-05-21 15:14:14 +09:00
transformer Avoid use_batch_norm as layers instance variable 2021-05-21 16:00:16 +09:00
utils Fix IPCBatchGenerator SIGINT handling 2021-04-27 17:50:40 +09:00
.gitignore Add mypy_cache to .gitignore 2020-10-05 11:27:53 +09:00
layers.py Avoid use_batch_norm as layers instance variable 2021-05-21 16:00:16 +09:00
residual.py Improve ResNet layers 2021-04-29 19:45:32 +09:00
train.py Trainer last summary fix + memory utils 2021-02-25 02:18:02 +09:00
trainer.py Pipeline implementation for BatchGenerator 2021-03-18 21:30:59 +09:00