From 9103c9878ef1a4a9ef9c4eba0013328bdeb566e5 Mon Sep 17 00:00:00 2001
From: alexcbb <alexchapin@hotmail.fr>
Date: Tue, 25 Jul 2023 10:25:43 +0200
Subject: [PATCH] Deactivate deterministic behavior
---
requirements.txt | 1 +
train_sa.py | 4 ++--
2 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/requirements.txt b/requirements.txt
index 49db6f2..39e8a42 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,3 +8,4 @@ matplotlib
tqdm
opencv-python
bitsandbytes
+segment_anything
diff --git a/train_sa.py b/train_sa.py
index a35c7c3..8e93e2c 100644
--- a/train_sa.py
+++ b/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)
--
GitLab