FlowFormer-Official icon indicating copy to clipboard operation
FlowFormer-Official copied to clipboard

Published 20 hours ago • verified publisher icon drinkingcoder

DDP training code and logging the metrics in wandb.

Open surajiitd opened this issue 1 year ago • 0 comments

Just made these changes:

  1. written DDP code for training and validation for faster training and validation.
  2. resolved bug in gma.py and encoders.py (these modules were not used while forward, so it was giving "all params not reduced after one step" error.
  3. written wandb logging code for training and validation metrics.

surajiitd avatar Feb 23 '24 11:02 surajiitd