LHeng

[email protected]

Know egg

Results 15 issues of


LHeng

loss

1

comment

Hello,I have a question about the loss. When I use the gem and rmac pooling in your code, the loss decreases slow, but I use AdaptiveAvgPool method in torchvision, loss...

question

RuntimeError: CUDA out of memory. Tried to allocate 12.00 MiB (GPU 0; 7.77 GiB total capacity; 5.54 GiB already allocated; 7.56 MiB free; 35.91 MiB cached)

您好，请问训练代码能开源吗

非常感谢

What is the class_weights int train.py

Could you please tell me what the class_weights mean and how to set it?

The target has dissappeared, but the model still has predicted result. How to solve it?

1

comment