Ldlbest
Ldlbest
> did you add __init__.py in the relevant folders? I added it to utils and bbox folders. yes,but it did‘t soloved.
> generated .pyd Please more details,thanks
> "python setup.py build", then copy the generated .pyd files into bbox folder. It works for me. Do I need to write a setup.py by myself?
lightning 需要在模型类中定义training_step函数，在training_step函数中进行反向传播，详见文档：https://lightning.ai/docs/pytorch/stable/model/train_model_basic.html
fp 16溢出，改成bf16就好了