updating data processing scripts to output to the correct location
This commit is contained in:
@@ -6,6 +6,4 @@
|
|||||||
# expert_class:str='NormalizedIntersimpleExpert',
|
# expert_class:str='NormalizedIntersimpleExpert',
|
||||||
# expert_args:dict={mu:0.001}):
|
# expert_args:dict={mu:0.001}):
|
||||||
|
|
||||||
cd ./src/data
|
python -m src.data.expert --locs='[DR_USA_Roundabout_FT]'
|
||||||
python -m expert --locs='[DR_USA_Roundabout_FT]'
|
|
||||||
cd ../
|
|
||||||
@@ -1 +1 @@
|
|||||||
from src.data.expert_trajectories import demonstrations, load_experts, process_experts
|
from src.data.expert import demonstrations, load_experts, process_experts
|
||||||
@@ -1,7 +1,7 @@
|
|||||||
import torch
|
import torch
|
||||||
from torch.utils.data import Dataset
|
from torch.utils.data import Dataset
|
||||||
import numpy as np
|
import numpy as np
|
||||||
from src.expert_data import load_expert_data
|
from src.data.expert_data import load_expert_data
|
||||||
import os
|
import os
|
||||||
opj = os.path.join
|
opj = os.path.join
|
||||||
|
|
||||||
|
|||||||
@@ -100,7 +100,7 @@ def save_video(env, expert):
|
|||||||
env.render()
|
env.render()
|
||||||
env.close()
|
env.close()
|
||||||
|
|
||||||
def load_experts(expert_files=[]):
|
def load_experts(expert_files):
|
||||||
"""
|
"""
|
||||||
Load expert trajectories from files and combine their transitions into a single RB
|
Load expert trajectories from files and combine their transitions into a single RB
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user