fix(wandb): ensure validation metrics logged at correct step
fix(wandb): ensure validation metrics logged at correct step

lilleswing/deepchem,miaecle/deepchem,peastman/deepchem,miaecle/deepchem,deepchem/deepchem,peastman/deepchem,lilleswing/deepchem,miaecle/deepchem,deepchem/deepchem,lilleswing/deepchem