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/eval_script
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/eval_script" have entirely different histories.
main ... cchi/eval_

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

2
conda_environment.yaml
View File

@ -46,8 +46,6 @@ dependencies:
- diffusers=0.11.1 - diffusers=0.11.1
- av=10.0.0 - av=10.0.0
- cmake=3.24.3 - cmake=3.24.3
# trick to avoid cpu affinity issue described in https://github.com/pytorch/pytorch/issues/99625
- llvm-openmp=14
# trick to force reinstall imagecodecs via pip # trick to force reinstall imagecodecs via pip
- imagecodecs==2022.8.8 - imagecodecs==2022.8.8
- pip: - pip:

2
conda_environment_real.yaml
View File

@ -46,8 +46,6 @@ dependencies:
- diffusers=0.11.1 - diffusers=0.11.1
- av=10.0.0 - av=10.0.0
- cmake=3.24.3 - cmake=3.24.3
# trick to avoid cpu affinity issue described in https://github.com/pytorch/pytorch/issues/99625
- llvm-openmp=14
# trick to force reinstall imagecodecs via pip # trick to force reinstall imagecodecs via pip
- imagecodecs==2022.8.8 - imagecodecs==2022.8.8
- pip: - pip:

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])