Deactivate deterministic behavior

requirements.txt

@@ -8,3 +8,4 @@ matplotlib
 tqdm
 opencv-python
 bitsandbytes
+segment_anything

train_sa.py

@@ -69,8 +69,8 @@ def main():
     )
 
     trainer = pl.Trainer(accelerator="gpu", devices=num_gpus, profiler="simple", 
-                         default_root_dir="./logs", logger=WandbLogger(project="slot-attention") if args.wandb else None,
-                         strategy="ddp" if num_gpus > 1 else "default", callbacks=[checkpoint_callback], deterministic=True,
+                         default_root_dir="./logs", logger=WandbLogger(project="slot-att") if args.wandb else None,
+                         strategy="ddp" if num_gpus > 1 else "default", callbacks=[checkpoint_callback],
                          log_every_n_steps=100, max_steps=num_train_steps)
 
     trainer.fit(model, train_loader, val_loader)