Skip to content

Commit 89dc851

Browse files
committed
create directory if needed
1 parent 35e0864 commit 89dc851

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

dae.py

+6
Original file line numberDiff line numberDiff line change
@@ -85,6 +85,12 @@ def create_network(network_spec, dataset):
8585

8686
@ex.capture
8787
def create_trainer(training, net_filename, verbose):
88+
import os
89+
import os.path
90+
dirname = os.path.dirname(net_filename)
91+
if not os.path.exists(dirname):
92+
os.makedirs(dirname)
93+
8894
trainer = bs.Trainer(bs.training.SgdStepper(training['learning_rate']),
8995
verbose=verbose)
9096
trainer.train_scorers = [bs.scorers.Hamming()]

0 commit comments

Comments
 (0)