8.12Sigma

ganchuanl

已于 2024-09-12 15:37:09 修改

阅读量412

点赞数 5

文章标签： python 开发语言

于 2024-08-12 15:26:11 首次发布

本文链接：https://blog.csdn.net/ganchuanl/article/details/141132888

版权

Traceback (most recent call last):
File "/home/gcl/project/cctrans-main/train.py", line 32, in <module>
    trainer.train()
File "/home/gcl/project/cctrans-main/train_helper_ALTGVT.py", line 214, in train
    self.train_epoch()
File "/home/gcl/project/cctrans-main/train_helper_ALTGVT.py", line 242, in train_epoch
    ot_loss, wd, ot_obj_value = self.ot_loss(
File "/home/gcl/anaconda3/envs/vmamba/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1190, in _call_impl
    return forward_call(*input, **kwargs)
File "/home/gcl/project/cctrans-main/losses/ot_loss.py", line 50, in forward
    P, log = sinkhorn(target_prob, source_prob, dis, self.reg, maxIter=self.num_of_iter_in_ot, log=True)
File "/home/gcl/project/cctrans-main/losses/bregman_pytorch.py", line 69, in sinkhorn
    return sinkhorn_knopp(a, b, C, reg, maxIter=maxIter,
File "/home/gcl/project/cctrans-main/losses/bregman_pytorch.py", line 146, in sinkhorn_knopp
    assert a.min() >= 0. and b.min() >= 0., 'Elements in a or b less than 0'
AssertionError: Elements in a or b less than 0