构建no-text数据集

2026-02-20 14:13:25 +08:00
parent ee257bcb6c
commit d85cce8a52
5 changed files with 194 additions and 62 deletions
--- a/utils.py
+++ b/utils.py
@@ -344,7 +344,7 @@ def load_data(dataset_dir, num_episodes, camera_names, batch_size_train, batch_s
        raise ValueError(f'Need at least 2 episodes for train/val split, found {len(episode_ids)} in {dataset_dir}')

    # obtain train test split
-    train_ratio = 0.8
+    train_ratio = 0.9
    shuffled_indices = np.random.permutation(len(episode_ids))
    train_count = int(train_ratio * len(episode_ids))
    train_indices = shuffled_indices[:train_count]