huangfu/InteractionImitation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Rename action to option

Browse Source
This commit is contained in:
Johannes Fischer
2021-09-13 16:43:45 +02:00
parent 9b8ceed9c9
commit 9c7e6cef3a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scratch/etienne/intersimple/gail_options_image.py
View File

@@ -154,7 +154,7 @@ def sample(env, generator, discriminator, level: str):
assert steps > 0 assert steps > 0
yield { yield {
'obs': obs, 'obs': obs,
'action': ch, 'option': ch,
'reward': r.detach() / steps, 'reward': r.detach() / steps,
'episode_start': episode_start, 'episode_start': episode_start,
'value': value.detach(), 'value': value.detach(),
@@ -182,7 +182,7 @@ def train_generator(env, generator, discriminator, num_samples):
for s in generator_samples[:-1]: for s in generator_samples[:-1]:
generator.rollout_buffer.add( generator.rollout_buffer.add(
obs=s['obs'], obs=s['obs'],
action=s['action'].cpu(), action=s['option'].cpu(),
reward=s['reward'].cpu(), reward=s['reward'].cpu(),
episode_start=s['episode_start'], episode_start=s['episode_start'],
value=s['value'], value=s['value'],