Skip to content
Snippets Groups Projects

Remove training info from DatasetManager

Merged Mélodie Boillet requested to merge remove-training-info into main
All threads resolved!
1 file
+ 1
2
Compare changes
  • Side-by-side
  • Inline
+ 1
2
@@ -139,7 +139,7 @@ class DatasetManager:
)
@staticmethod
def seed_worker(worker_id):
def seed_worker():
worker_seed = torch.initial_seed() % 2**32
np.random.seed(worker_seed)
random.seed(worker_seed)
@@ -232,7 +232,6 @@ class GenericDataset(Dataset):
self.params["config"]["padding_value"] = self.padding_value
self.curriculum_config = None
self.training_info = None
def __len__(self):
return len(self.samples)
Loading