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