whlgstone2

train one-class dataset error, After following the instructions for custom datasets training. waiting for your advice. Thanks

1

[rank1]: File "/root/download/D-FINE/src/zoo/dfine/hybrid_encoder.py", line 267, in forward [rank1]: src, _ = self.self_attn(q, k, value=src, attn_mask=src_mask) [rank1]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ [rank1]: File "/root/miniconda3/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1736, in _wrapped_call_impl [rank1]: return self._call_impl(*args, **kwargs) [rank1]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^...