small

2023-03-14 14:04:08 -07:00
parent 5a33ee8db0
commit 76cf30b4fe
3 changed files with 4 additions and 3 deletions
--- a/utils.py
+++ b/utils.py
@@ -109,6 +109,7 @@ def get_norm_stats(dataset_dir, num_episodes):


 def load_data(dataset_dir, num_episodes, camera_names, batch_size_train, batch_size_val):
+    print(f'\nData from: {dataset_dir}\n')
    # obtain train test split
    train_ratio = 0.8
    shuffled_indices = np.random.permutation(num_episodes)