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

Cleanup

Browse Source
This commit is contained in:
Johannes Fischer
2021-10-28 13:47:54 +02:00
parent 8703b11dee
commit c5b043c49f
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
scratch/johannes/gail_options_image.py
View File

@@ -5,10 +5,9 @@
from src.discriminator import CnnDiscriminator, CnnDiscriminatorFlatAction from src.discriminator import CnnDiscriminator, CnnDiscriminatorFlatAction
from imitation.algorithms import adversarial from imitation.algorithms import adversarial
import stable_baselines3 import stable_baselines3
from stable_baselines3.common.evaluation import evaluate_policy
import torch.utils.data import torch.utils.data
import numpy as np import numpy as np
from intersim.envs.intersimple import Intersimple, NRasterized, NRasterizedInfo, NRasterizedIncrementingAgent, NRasterizedRandomAgent, speed_reward from intersim.envs.intersimple import NRasterized, NRasterizedRandomAgent, speed_reward
import itertools import itertools
import functools import functools
from torch.distributions import Categorical from torch.distributions import Categorical