[memory leak
] Replace GradientState -> DataLoader reference with weakrefs
#5444
Loading
memory leak
] Replace GradientState -> DataLoader reference with weakrefs
#5444