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

changing relative state dim to 6

Browse Source
This commit is contained in:
Arec
2021-07-29 02:54:41 -07:00
parent fdbdc7f9f0
commit e47d69dbc1
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
config/networks.json5
View File

@@ -6,7 +6,7 @@
output_dim: 5 output_dim: 5
}, },
deepsets: { deepsets: {
input_dim: 5, // number of relative state vars for others input_dim: 6, // number of relative state vars for others
phi: { phi: {
hidden_n: 2, hidden_n: 2,
hidden_dim: 20, hidden_dim: 20,

4
src/bc/bc.py
View File

@@ -14,7 +14,7 @@ def bc_config(ray_config):
config = { config = {
'ego_encoder': {'input_dim': 5, 'hidden_n': 0, 'hidden_dim':0, 'output_dim': 0}, 'ego_encoder': {'input_dim': 5, 'hidden_n': 0, 'hidden_dim':0, 'output_dim': 0},
'deepsets': { 'deepsets': {
'input_dim': 5, 'input_dim': 6,
'phi': { 'phi': {
'hidden_n': ray_config['deepsets_phi_hidden_n'], 'hidden_n': ray_config['deepsets_phi_hidden_n'],
'hidden_dim': ray_config['deepsets_phi_hidden_dim'] 'hidden_dim': ray_config['deepsets_phi_hidden_dim']
@@ -39,7 +39,7 @@ def bc_config(ray_config):
'lr':ray_config['lr'], 'lr':ray_config['lr'],
'weight_decay':ray_config['weight_decay'] 'weight_decay':ray_config['weight_decay']
}, },
'train_epochs': 50, 'train_epochs': 20,
'train_batch_size': ray_config['train_batch_size'], 'train_batch_size': ray_config['train_batch_size'],
'loss': ray_config['loss'], 'loss': ray_config['loss'],