def test_default(self):
args = tools.AttrDict(logdir=self.create_tempdir(),
num_runs=1,
config='debug',
params=tools.AttrDict(task='cheetah_run',
train_steps=10,
test_steps=10,
max_steps=50,
batch_size=(5, 10)),
ping_every=0,
resume_runs=False)
tf.app.run(lambda _: train.main(args), [sys.argv[0]])
def test_dm_control_isolate_thread(self):
args = tools.AttrDict(
logdir=self.get_temp_dir(),
num_runs=1,
config='debug',
params=tools.AttrDict(
task='cup_catch',
isolate_envs='thread',
max_steps=30),
ping_every=0,
resume_runs=False)
try:
tf.app.run(lambda _: train.main(args), [sys.argv[0]])
except SystemExit:
pass
def test_dummy_isolate_process(self):
args = tools.AttrDict(
logdir=self.get_temp_dir(),
num_runs=1,
config='debug',
params=tools.AttrDict(
task='dummy',
isolate_envs='process',
max_steps=30),
ping_every=0,
resume_runs=False)
try:
tf.app.run(lambda _: train.main(args), [sys.argv[0]])
except SystemExit:
pass