You could either directly calculate the mean
and std
in case you are preloading the data or you could iterate the DataLoader
and calculate these stats from each batch as seen in e.g. this post.
You could either directly calculate the mean
and std
in case you are preloading the data or you could iterate the DataLoader
and calculate these stats from each batch as seen in e.g. this post.