+ all_actions += [actions[None, :]]
+ return torch.cat(all_frames, 0).contiguous(), torch.cat(all_actions, 0)
+
+
+def create_data_and_processors(
+ nb_train_samples,
+ nb_test_samples,
+ mode,
+ nb_steps,
+ nb_epochs=10,
+ device=torch.device("cpu"),
+ device_storage=torch.device("cpu"),
+ logger=None,
+):
+ assert mode in ["first_last"]