adding tool for visualizing acceleration distributions, and making nframes an arg

2021-07-29 07:35:21 -07:00
parent afc3719ab9
commit 3b4ef6ffb5
4 changed files with 44 additions and 9 deletions
--- a/src/bc/bc.py
+++ b/src/bc/bc.py
@@ -39,7 +39,7 @@ def bc_config(ray_config):
            'lr':ray_config['lr'],
            'weight_decay':ray_config['weight_decay']
        },
-        'train_epochs': 20,
+        'train_epochs': 40,
        'train_batch_size': ray_config['train_batch_size'],
        'loss': ray_config['loss'],