match weights name to github release

compphoto · Jan 4, 2024 · 2d19fe4 · 2d19fe4
1 parent 6b2326d
commit 2d19fe4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/intrinsic_compositing/albedo/pipeline.py b/intrinsic_compositing/albedo/pipeline.py
@@ -152,7 +152,7 @@ def load_albedo_harmonizer():
     args.lr_editnet = 0.00001
     args.batch_size = 1
 
-    args.checkpoint_load_path = f'{CACHE_PATH}/albedo_harmonization/168000_net_Parameters.pth'
+    args.checkpoint_load_path = f'{CACHE_PATH}/albedo_harmonization/albedo_paper_weights.pth'
     # args.checkpoint_load_path = f'{cur_path}/checkpoints/168000_net_Parameters.pth'
 
     if not os.path.exists(args.checkpoint_load_path):