请问LOSS是怎么反向传播的？
a897456 opened this issue · 1 comments
a897456 commented
loss是在training_step和validation_step产生，但是在main（）中好像没有loss的字眼。请问是怎么back的？
ldlbest commented
lightning 需要在模型类中定义training_step函数，在training_step函数中进行反向传播，详见文档：https://lightning.ai/docs/pytorch/stable/model/train_model_basic.html