pytorch中一些API的功能

1、transforms.ToTensor()(img)
该句代码会将原本(128,128)的img(PIL读取的)转换成(1,128,128),同时像素也会被归一化
2、tqdm与print打印出的信息乱序
解决办法:tqdm(enumerate(train_dataloader), file=sys.stdout)-使tqdm的打印管道与print的一致

posted @ 2022-04-11 08:55  原来是只呆燕  阅读(45)  评论(0)    收藏  举报