Spaces:
Runtime error
Runtime error
loop_smpl bug of PIFu
Browse files- apps/infer.py +3 -4
apps/infer.py
CHANGED
@@ -153,10 +153,9 @@ def generate_model(in_path, model_type):
|
|
153 |
|
154 |
# smpl optimization
|
155 |
|
156 |
-
loop_smpl = tqdm(
|
157 |
-
range(config_dict['loop_smpl'] if cfg.net.prior_type != "pifu" else 1))
|
158 |
|
159 |
-
for
|
160 |
|
161 |
optimizer_smpl.zero_grad()
|
162 |
|
@@ -371,7 +370,7 @@ def generate_model(in_path, model_type):
|
|
371 |
|
372 |
loop_cloth = tqdm(range(config_dict['loop_cloth']))
|
373 |
|
374 |
-
for
|
375 |
|
376 |
optimizer_cloth.zero_grad()
|
377 |
|
|
|
153 |
|
154 |
# smpl optimization
|
155 |
|
156 |
+
loop_smpl = tqdm(range(config_dict['loop_smpl']))
|
|
|
157 |
|
158 |
+
for _ in loop_smpl:
|
159 |
|
160 |
optimizer_smpl.zero_grad()
|
161 |
|
|
|
370 |
|
371 |
loop_cloth = tqdm(range(config_dict['loop_cloth']))
|
372 |
|
373 |
+
for _ in loop_cloth:
|
374 |
|
375 |
optimizer_cloth.zero_grad()
|
376 |
|