holy fuking shit

2026-09-12 01:56:34 +08:00
parent 36f354736c
commit d975309a81
83 changed files with 2748 additions and 896 deletions
@@ -82,7 +82,6 @@ def bc_validation_mask(episode_ids, *, seed: int):

 def train_bc(x, y, episode_ids, *, epochs: int, batch_size: int, lr: float,
             device: str, seed: int):
-    import numpy as np
    import torch
    BCMlp = _bc_module(LOW_OBS_DIM, LOW_NUM_ACTIONS, HIDDEN)
    model = BCMlp().to(device)