hongshaorou/diffusion_policy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Compare commits

base: hongshaorou:main
Branches Tags
hongshaorou:main
hongshaorou:cchi/fix_cpu_affinity
hongshaorou:cchi/eval_script
hongshaorou:cchi/fix_transformer_impainting
hongshaorou:cchi/bug_fix_unet1d
hongshaorou:cchi/bug_fix_eval_sample
..
compare: hongshaorou:cchi/fix_cpu_affinity
Branches Tags
hongshaorou:main
hongshaorou:cchi/fix_cpu_affinity
hongshaorou:cchi/eval_script
hongshaorou:cchi/fix_transformer_impainting
hongshaorou:cchi/bug_fix_unet1d
hongshaorou:cchi/bug_fix_eval_sample

No commits in common. "main" and "cchi/fix_cpu_affinity" have entirely different histories.
main ... cchi/fix_c

1 changed files with 2 additions and 2 deletions
Show Stats Expand all files Collapse all files

4
diffusion_policy/model/common/rotation_transformer.py
View File

@ -40,7 +40,7 @@ class RotationTransformer:
getattr(pt, f'matrix_to_{from_rep}') getattr(pt, f'matrix_to_{from_rep}')
] ]
if from_convention is not None: if from_convention is not None:
funcs = [functools.partial(func, convention=from_convention) funcs = [functools.partial(func, convernsion=from_convention)
for func in funcs] for func in funcs]
forward_funcs.append(funcs[0]) forward_funcs.append(funcs[0])
inverse_funcs.append(funcs[1]) inverse_funcs.append(funcs[1])
@ -51,7 +51,7 @@ class RotationTransformer:
getattr(pt, f'{to_rep}_to_matrix') getattr(pt, f'{to_rep}_to_matrix')
] ]
if to_convention is not None: if to_convention is not None:
funcs = [functools.partial(func, convention=to_convention) funcs = [functools.partial(func, convernsion=to_convention)
for func in funcs] for func in funcs]
forward_funcs.append(funcs[0]) forward_funcs.append(funcs[0])
inverse_funcs.append(funcs[1]) inverse_funcs.append(funcs[1])