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

Fix imports

Browse Source
This commit is contained in:
ebuehrle
2022-02-18 10:18:10 +01:00
parent 84351e77f2
commit ce22516893
2 changed files with 20 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
scratch/etienne/trpo/experiments/sgail-options-setobs2.py
View File

@@ -1,19 +1,22 @@
# %% # %%
import sys
sys.path.append('../../../../')
import gym import gym
from safe_options.options import gail from src.safe_options.options import gail
from core.gail import Buffer from src.core.gail import Buffer
from core.value import SetValue from src.core.value import SetValue
from safe_options.policy import SetMaskedDiscretePolicy from src.safe_options.policy import SetMaskedDiscretePolicy
from core.discriminator import DeepsetDiscriminator from src.core.discriminator import DeepsetDiscriminator
import torch.optim import torch.optim
from intersim.envs import IntersimpleLidarFlatRandom from intersim.envs import IntersimpleLidarFlatRandom
from intersim.envs.intersimple import speed_reward from intersim.envs.intersimple import speed_reward
import functools import functools
from util.wrappers import CollisionPenaltyWrapper, TransformObservation, Setobs from src.util.wrappers import CollisionPenaltyWrapper, TransformObservation, Setobs
import numpy as np import numpy as np
from safe_options.options import SafeOptionsEnv from src.safe_options.options import SafeOptionsEnv
from torch.utils.tensorboard import SummaryWriter from torch.utils.tensorboard import SummaryWriter
from core.reparam_module import ReparamPolicy from src.core.reparam_module import ReparamPolicy
obs_min = np.array([ obs_min = np.array([
[-1000, -1000, 0, -np.pi, -1e-1, 0.], [-1000, -1000, 0, -np.pi, -1e-1, 0.],

15
scratch/etienne/trpo/experiments/sgail-ppo-options-setobs2.py
View File

@@ -1,16 +1,19 @@
# %% # %%
import sys
sys.path.append('../../../../')
import gym import gym
from safe_options.options import gail_ppo, Buffer from src.safe_options.options import gail_ppo, Buffer
from core.value import SetValue from src.core.value import SetValue
from safe_options.policy import SetMaskedDiscretePolicy from src.safe_options.policy import SetMaskedDiscretePolicy
from core.discriminator import DeepsetDiscriminator from src.core.discriminator import DeepsetDiscriminator
import torch.optim import torch.optim
from intersim.envs import IntersimpleLidarFlatRandom from intersim.envs import IntersimpleLidarFlatRandom
from intersim.envs.intersimple import speed_reward from intersim.envs.intersimple import speed_reward
import functools import functools
from util.wrappers import CollisionPenaltyWrapper, TransformObservation, Setobs from src.util.wrappers import CollisionPenaltyWrapper, TransformObservation, Setobs
import numpy as np import numpy as np
from safe_options.options import SafeOptionsEnv from src.safe_options.options import SafeOptionsEnv
from torch.utils.tensorboard import SummaryWriter from torch.utils.tensorboard import SummaryWriter
obs_min = np.array([ obs_min = np.array([