如何在pytorch中计算dataloader的mean和std

如何在pytorch中计算dataloader的mean和std

data = next(iter(train_loader))

mean = data[0].mean()

std = data[0].std()

 

-----------

came from 

https://www.youtube.com/watch?v=bCQ2cNhUWQ8&t=58s&ab_channel=deeplizard

 

 

posted @ 2022-03-20 21:36  bH1pJ  阅读(116)  评论(0)  编辑  收藏  举报