bug fix

MarcusLoppe · Jun 22, 2024 · 9e1bb0d · 9e1bb0d
1 parent 3e7ed57
commit 9e1bb0d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/meshgpt_pytorch/trainer.py b/meshgpt_pytorch/trainer.py
@@ -126,6 +126,7 @@ def __init__(
 
         self.dataloader = DataLoader(
             dataset,
+            shuffle = True,
             batch_size = batch_size, 
             drop_last = True,
             collate_fn = partial(custom_collate, pad_id = model.pad_id)
@@ -141,6 +142,7 @@ def __init__(
 
             self.val_dataloader = DataLoader(
                 val_dataset,
+                shuffle = True,
                 batch_size = batch_size, 
                 drop_last = True,
                 collate_fn = partial(custom_collate, pad_id = model.pad_id)
@@ -468,6 +470,7 @@ def __init__(
 
         self.dataloader = DataLoader(
             dataset,
+            shuffle = True,
             batch_size = batch_size, 
             drop_last = True,
             collate_fn = partial(custom_collate, pad_id = model.pad_id)
@@ -483,6 +486,7 @@ def __init__(
 
             self.val_dataloader = DataLoader(
                 val_dataset,
+                shuffle = True,
                 batch_size = batch_size, 
                 drop_last = True,
                 collate_fn = partial(custom_collate, pad_id = model.pad_id)