Skip to content

Commit f5d3942

Browse files
committed
split before declaring file name
1 parent 283249d commit f5d3942

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

modules/hypernetworks/hypernetwork.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -382,6 +382,7 @@ def train_hypernetwork(hypernetwork_name, learn_rate, batch_size, data_root, log
382382
shared.state.textinfo = "Initializing hypernetwork training..."
383383
shared.state.job_count = steps
384384

385+
hypernetwork_name = hypernetwork_name.rsplit('(', 1)[0]
385386
filename = os.path.join(shared.cmd_opts.hypernetwork_dir, f'{hypernetwork_name}.pt')
386387

387388
log_directory = os.path.join(log_directory, datetime.datetime.now().strftime("%Y-%m-%d"), hypernetwork_name)
@@ -393,7 +394,6 @@ def train_hypernetwork(hypernetwork_name, learn_rate, batch_size, data_root, log
393394
else:
394395
hypernetwork_dir = None
395396

396-
hypernetwork_name = hypernetwork_name.rsplit('(', 1)[0]
397397
if create_image_every > 0:
398398
images_dir = os.path.join(log_directory, "images")
399399
os.makedirs(images_dir, exist_ok=True)

0 commit comments

Comments
 (0)