FastVideo
FastVideo copied to clipboard
Duplicated code in the training pipeline.
I think the following codes should be removed, right?
validation_batch.total_loss = validation_batch.total_loss
validation_batch.grad_norm = validation_batch.grad_norm
https://github.com/hao-ai-lab/FastVideo/blob/9ce7c8039e3fec4c632b8d29a2e41e418a9b56d6/fastvideo/training/training_pipeline.py#L522
Training part is used here? Validation part could be removed.
But the following part in the training code still seems duplicated:
training_batch.total_loss = training_batch.total_loss
training_batch.grad_norm = training_batch.grad_norm
ref to here.