Skip to content

Commit 97a8792

Browse files
author
donglaiw
committed
create folder
1 parent b31c68c commit 97a8792

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

connectomics/engine/trainer.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -321,6 +321,9 @@ def test_singly(self):
321321
for i in ran:
322322
self.test_filename = output_name[i]
323323
if not os.path.exists(self.test_filename):
324+
output_dir = os.path.basename(self.test_filename)
325+
if not os.path.exists(output_dir):
326+
os.makedirs(output_dir)
324327
if dir_name is not None:
325328
# directly load from dir_name_init and img_name_init
326329
dataset = get_dataset(

0 commit comments

Comments
 (0)