From 50a951d065b80da496da5d360024f5874b125a35 Mon Sep 17 00:00:00 2001
From: Sukjun Hwang <sukjunh@andrew.cmu.edu>
Date: Wed, 17 Jul 2024 13:04:00 -0400
Subject: [PATCH] correct sampler if repeat augment

---
 configs/experiment/s4nd/vit/vit_b_16_imagenet.yaml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/configs/experiment/s4nd/vit/vit_b_16_imagenet.yaml b/configs/experiment/s4nd/vit/vit_b_16_imagenet.yaml
index 6a63ab1..f3e4519 100644
--- a/configs/experiment/s4nd/vit/vit_b_16_imagenet.yaml
+++ b/configs/experiment/s4nd/vit/vit_b_16_imagenet.yaml
@@ -28,6 +28,7 @@ trainer:
   max_epochs: 310
   precision: 16
   devices: 8
+  use_distributed_sampler: ${eval:"${dataset.num_aug_repeats} == 0"}  # only true if using RepeatAug
 
 train:
   seed: 1112