Skip to content

预训练权重以及报错 #3

Description

@Joycce516

您好,请问可以重新分享一下DLA-34 COCO pretranined CenterNet model weight的权重文件吗,链接失效了,目前我只能通过网上找到的其他DLA-34 COCO预训练权重进行测试;
除此之外,我在训练时出现了以下报错:
Traceback (most recent call last):
File "F:\My project\OUTrack-main\src\train.py", line 162, in
main(opt)
File "F:\My project\OUTrack-main\src\train.py", line 121, in main
log_dict_train, _ = trainer.train(epoch, train_loader)
File "F:\My project\OUTrack-main\src\lib\trains\base_trainer.py", line 205, in train
return self.run_epoch('train', epoch, data_loader)
File "F:\My project\OUTrack-main\src\lib\trains\base_trainer.py", line 148, in run_epoch
output, loss, loss_stats = model_with_loss(batch, batch_pre)
File "D:\anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 1130, in _call_impl
return forward_call(*input, **kwargs)
File "F:\My project\OUTrack-main\src\lib\trains\base_trainer.py", line 66, in forward
loss, loss_stats = self.loss(outputs, batch, outputs_pre, batch_pre)
File "D:\anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 1130, in _call_impl
return forward_call(*input, **kwargs)
File "F:\My project\OUTrack-main\src\lib\trains\mot.py", line 63, in forward
output['hm'] = _sigmoid(output['hm'])
File "F:\My project\OUTrack-main\src\lib\models\utils.py", line 9, in sigmoid
y = torch.clamp(x.sigmoid
(), min=1e-4, max=1-1e-4)
RuntimeError: Output 0 of SplitBackward0 is a view and is being modified inplace. This view is the output of a function that returns multiple views. Such functions do not allow the output views to be modified inplace. You should replace the inplace operation by an out-of-place.
不知道您是否遇到过类似报错(也有可能是因为权重文件不匹配导致的),希望能得到您的回复!再次感谢!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions