update mpi usage
This commit is contained in:
@@ -40,8 +40,8 @@ def main():
|
|||||||
|
|
||||||
# Command line arguments
|
# Command line arguments
|
||||||
args = sys.argv[1:]
|
args = sys.argv[1:]
|
||||||
|
enable_draw = False
|
||||||
world = build_world(args, enable_draw=False)
|
world = build_world(args, enable_draw)
|
||||||
|
|
||||||
run()
|
run()
|
||||||
shutdown()
|
shutdown()
|
||||||
|
|||||||
@@ -15,6 +15,7 @@ from pybullet_envs.deep_mimic.env.pybullet_deep_mimic_env import PyBulletDeepMim
|
|||||||
import sys
|
import sys
|
||||||
import random
|
import random
|
||||||
|
|
||||||
|
update_timestep = 1./600.
|
||||||
|
|
||||||
def update_world(world, time_elapsed):
|
def update_world(world, time_elapsed):
|
||||||
timeStep = 1./600.
|
timeStep = 1./600.
|
||||||
@@ -43,8 +44,9 @@ args = sys.argv[1:]
|
|||||||
|
|
||||||
|
|
||||||
|
|
||||||
def build_world(args, enable_draw, playback_speed=1):
|
def build_world(args, enable_draw):
|
||||||
arg_parser = build_arg_parser(args)
|
arg_parser = build_arg_parser(args)
|
||||||
|
print("enable_draw=",enable_draw)
|
||||||
env = PyBulletDeepMimicEnv(args, enable_draw)
|
env = PyBulletDeepMimicEnv(args, enable_draw)
|
||||||
world = RLWorld(env, arg_parser)
|
world = RLWorld(env, arg_parser)
|
||||||
#world.env.set_playback_speed(playback_speed)
|
#world.env.set_playback_speed(playback_speed)
|
||||||
@@ -72,6 +74,7 @@ def build_world(args, enable_draw, playback_speed=1):
|
|||||||
world.reset()
|
world.reset()
|
||||||
return world
|
return world
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
world = build_world(args, True)
|
world = build_world(args, True)
|
||||||
while (world.env._pybullet_client.isConnected()):
|
while (world.env._pybullet_client.isConnected()):
|
||||||
timeStep = 1./600.
|
timeStep = 1./600.
|
||||||
|
|||||||
Reference in New Issue
Block a user