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

Sample with replacement

Browse Source
This commit is contained in:
ebuehrle
2021-11-04 19:54:14 +01:00
parent f794a96112
commit 081fb4e6ab
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/options2.py
View File

@@ -3,7 +3,6 @@ import torch
from src.util.collisions import feasible from src.util.collisions import feasible
import numpy as np import numpy as np
from collections import deque from collections import deque
import itertools
def imitation_discriminator(discriminator): def imitation_discriminator(discriminator):
return lambda obs, action, next_obs, done: discriminator.discrim_net.predict_reward_train( return lambda obs, action, next_obs, done: discriminator.discrim_net.predict_reward_train(
@@ -82,7 +81,8 @@ class OptionsEnv(gym.Wrapper):
def sample_ll(self, n): def sample_ll(self, n):
assert n <= self.ll_buffer_capacity, f'Sample size of {n} exceeds buffer capacity of {self.ll_buffer_capacity}' assert n <= self.ll_buffer_capacity, f'Sample size of {n} exceeds buffer capacity of {self.ll_buffer_capacity}'
assert n <= len(self.ll_buffer), f'Sample size of {n} exceeds buffer size of {len(self.ll_buffer)}' assert n <= len(self.ll_buffer), f'Sample size of {n} exceeds buffer size of {len(self.ll_buffer)}'
return list(itertools.islice(self.ll_buffer, n)) ind = np.random.randint(len(self.ll_buffer), size=n)
return list(self.ll_buffer[i] for i in ind)
class RenderOptions(OptionsEnv): class RenderOptions(OptionsEnv):