How can I synchronization metrics in `ray.train` valid_loop

Was this solved in How to make all use of the GPU memory in Ray.tune - #3 by Yard1?