17 lines
476 B
Python
17 lines
476 B
Python
import torch
|
|
import pickle
|
|
import json5
|
|
from src.bc import BehaviorCloningPolicy
|
|
config_path = "config/networks.json5"
|
|
|
|
with open(config_path, 'r') as cfg:
|
|
config = json5.load(cfg)
|
|
|
|
filestr = 'tests/policies/base_'
|
|
|
|
|
|
batch = pickle.load(open(filestr+'_test_batch.pkl', 'rb'))
|
|
batch['relative_state'] = batch['relative_state'].float()
|
|
policy = BehaviorCloningPolicy.load_model(config, filestr)
|
|
policy.policy = policy.policy.type(batch['state'].dtype)
|
|
action = policy(batch) |