We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 9593115 commit 4b2ee64Copy full SHA for 4b2ee64
1 file changed
examples/torch_model_loading/torch_load.chpl
@@ -14,7 +14,7 @@ proc main(args: [] string) {
14
image = image.resize(224,224);
15
writeln("Resized image: ", image.shape);
16
17
- var batchedImage = image.unsqueeze(0);
+ var batchedImage = ndarray.loadFrom(args[1],3,real(32)).unsqueeze(0);
18
writeln("Batched image: ", batchedImage.shape);
19
20
batchedImage = batchedImage.resize(224,224);
0 commit comments