jianght commited on
Commit
3e4d5b8
·
verified ·
1 Parent(s): f982020

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -40,8 +40,8 @@ model = model.eval().to("cuda")
40
  transform = transforms.Compose([
41
  transforms.ToTensor(),
42
  transforms.Normalize(
43
- mean=(0.485, 0.456, 0.406),
44
- std=(0.229, 0.224, 0.225),
45
  ),
46
  ])
47
 
 
40
  transform = transforms.Compose([
41
  transforms.ToTensor(),
42
  transforms.Normalize(
43
+ mean=(0.485, 0.456, 0.406),
44
+ std=(0.229, 0.224, 0.225),
45
  ),
46
  ])
47