noooo

alex_trainL.py

@@ -80,7 +80,7 @@ def train_model(args, model_id, dataset_id):
         # List full of individual sequence dataset objects. Iterate through sequence names creating datasets.
         seq_datasets = []
         for sample in seq_names:
-            #print(sample)
+            print(sample)
             # Take a smaller spatial crop to get more granularity in the temporal dimension. (same crop as in E-RAFT)
             seq_datasets.append(SequenceEDENN(Path(root_folder),sample,num_bins=args.n_bins, crop_window=(288, 384), mode="val"))
 
@@ -90,6 +90,7 @@ def train_model(args, model_id, dataset_id):
         val_data = ConcatDataset(seq_datasets[:1])
 
         print(train_data)
+        print(val_data)
 
         train_loader = DataLoader(train_data, batch_size=args.batch_size, num_workers=1, shuffle= False, pin_memory = True)
         val_loader = DataLoader(train_data, batch_size=args.batch_size, num_workers=1, shuffle = False, pin_memory = True)

edenn/datasets/dsec.py

@@ -201,10 +201,7 @@ class SequenceEDENN(Dataset):
         print(self.flow_names[index])
         print(self.flow_root)
         ts_start = self.flow_timestamps[index][0]
-        ts_end = self.flow_timestamps[index][1]
-
-        print(ts_start)
-        print(ts_end)
+        ts_end =  self.flow_timestamps[index][1]
 
         # Get specific event slice
         event_data = self.event_slicer.get_events(ts_start, ts_end)