我这个说'BalancedDataParallel' object has no attribute 'model' 请问知道是什么原因吗

[Sundia-cs]

torch.nn.modules.module.ModuleAttributeError: 'BalancedDataParallel' object has no attribute 'model'

[Sundia-cs]

raceback (most recent call last):
File "train_cstrack.py", line 522, in
train(hyp, opt, device, tb_writer)
File "train_cstrack.py", line 328, in train
loss, loss_items = mot_loss(pred, targets.to(device), model) # scaled by batch_size
File "/data/amax/b510/cs/.conda/envs/CSTrack/lib/python3.8/site-packages/torch/nn/modules/module.py", line 722, in _call_impl
result = self.forward(*input, **kwargs)
File "/data/amax/b510/cs/SOTS/CSTrack/tracking/../lib/core/mot/base_trainer.py", line 319, in forward
tcls, tbox, indices, anchors, indices_id, tids = build_targets(p, targets, model) # targets
File "/data/amax/b510/cs/SOTS/CSTrack/tracking/../lib/core/mot/base_trainer.py", line 227, in build_targets
det = model.module.model[-1] if is_parallel(model) else model.model[-1] # Detect() module
File "/data/amax/b510/cs/.conda/envs/CSTrack/lib/python3.8/site-packages/torch/nn/modules/module.py", line 771, in getattr
raise ModuleAttributeError("'{}' object has no attribute '{}'".format(
torch.nn.modules.module.ModuleAttributeError: 'BalancedDataParallel' object has no attribute 'model'