source code
def anim(i):
# update SMBLD
cur_beta_idx, cur_step = i // num_steps, i % num_steps
val = shape_range[cur_step]
mesh.multi_betas[0, cur_beta_idx] = val # Update betas
fig.suptitle(f"{name.title()}\nS{cur_beta_idx} : {val:+.2f}", fontsize=50) # update text
return dict(mesh=mesh.get_meshes(), equalize=False)
Modified code
Add with torch.no_grad():
will be OK!
def anim(i):
# update SMBLD
cur_beta_idx, cur_step = i // num_steps, i % num_steps
val = shape_range[cur_step]
#print("\ncur_beta_idx:",cur_beta_idx,mesh.multi_betas[0, cur_beta_idx])
with torch.no_grad():###添加
mesh.multi_betas[0, cur_beta_idx] = val # Update betas
fig.suptitle(f"{name.title()}\nS{cur_beta_idx} : {val:+.2f}", fontsize=50) # update text
return dict(mesh=mesh.get_meshes(), equalize=False)