Update README.md

README.md

@@ -76,7 +76,7 @@ to_idx   = dataset.meta.episodes["dataset_to_index"][episode_index]
 frame_index = from_idx
 frame = dict(dataset[frame_index])
 
-batch = preprocess(sample)
+batch = preprocess(frame)
 with torch.inference_mode():
     pred_action = policy.select_action(batch)
     # use your policy postprocess, this post process the action