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

making saving and loading functions class requirements, working on behavior cloning policy class and training function

Browse Source
This commit is contained in:
Arec
2021-07-20 08:29:00 -07:00
parent 1a74fa5237
commit 6794b4cad8
4 changed files with 110 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/__init__.py
View File

@@ -1,2 +1,3 @@
from src.expert_data import generate_expert_data, load_expert_data
from src.data_utils import InteractionDatasetSingleAgent
from src.data_utils import InteractionDatasetSingleAgent
from src.metrics import metrics