corentin/torch_utils
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Trainer implementation

Browse source 
* Add Deconv2d
* Fix BatchGenerator save option when using current directory
This commit is contained in:
Corentin 2020-12-25 15:50:38 +09:00
commit f2282e3216
3 changed files with 263 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

2
utils/batch_generator.py
View file

@ -31,7 +31,7 @@ class BatchGenerator:
if save is not None:
if '.' not in os.path.basename(save_path):
save_path += '.hdf5'
if not os.path.exists(os.path.dirname(save_path)):
if os.path.dirname(save_path) and not os.path.exists(os.path.dirname(save_path)):
os.makedirs(os.path.dirname(save_path))
if save and os.path.exists(save_path):