Attrubte error showing up when tryig to integrate new loss function

[tejasri19]

Starting training for 2 epochs...

  Epoch    GPU_mem   box_loss   cls_loss   dfl_loss  Instances       Size

0%| | 0/78 [00:00<?, ?it/s]
Traceback (most recent call last):
File "/home/tejasri/praneela/yolov10/train.py", line 189, in
trainer.train()
File "/home/tejasri/praneela/yolov10/ultralytics/engine/trainer.py", line 213, in train
self._do_train(world_size)
File "/home/tejasri/praneela/yolov10/ultralytics/engine/trainer.py", line 363, in _do_train
for i, batch in pbar:
File "/home/tejasri/anaconda3/envs/yolov10/lib/python3.9/site-packages/tqdm/std.py", line 1181, in iter
for obj in iterable:
File "/home/tejasri/praneela/yolov10/ultralytics/data/build.py", line 49, in iter
yield next(self.iterator)
File "/home/tejasri/anaconda3/envs/yolov10/lib/python3.9/site-packages/torch/utils/data/dataloader.py", line 633, in next
data = self._next_data()
File "/home/tejasri/anaconda3/envs/yolov10/lib/python3.9/site-packages/torch/utils/data/dataloader.py", line 1345, in _next_data
return self._process_data(data)
File "/home/tejasri/anaconda3/envs/yolov10/lib/python3.9/site-packages/torch/utils/data/dataloader.py", line 1371, in _process_data
data.reraise()
File "/home/tejasri/anaconda3/envs/yolov10/lib/python3.9/site-packages/torch/_utils.py", line 644, in reraise
raise exception
AttributeError: Caught AttributeError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/home/tejasri/anaconda3/envs/yolov10/lib/python3.9/site-packages/torch/utils/data/_utils/worker.py", line 308, in _worker_loop
data = fetcher.fetch(index)
File "/home/tejasri/anaconda3/envs/yolov10/lib/python3.9/site-packages/torch/utils/data/_utils/fetch.py", line 54, in fetch
return self.collate_fn(data)
File "/home/tejasri/praneela/yolov10/ultralytics/data/dataset.py", line 211, in collate_fn
keys = batch[0].keys()
AttributeError: 'NoneType' object has no attribute 'keys

Please share how to work with new loss functions

@jameslahm @leonnil