diffusion_policy_gbc / env_runner /real_pusht_image_runner.py
ducido's picture
Initial commit
987ed1b verified
Raw
History Blame Contribute Delete
346 Bytes
from diffusion_policy.policy.base_image_policy import BaseImagePolicy
from diffusion_policy.env_runner.base_image_runner import BaseImageRunner
class RealPushTImageRunner(BaseImageRunner):
def __init__(self,
output_dir):
super().__init__(output_dir)
def run(self, policy: BaseImagePolicy):
return dict()