Update load_model.cs

Глеб Брыкин · web-flow · commit 0cb6b393da0b · 2021-03-23T11:53:57.000+03:00
diff --git a/Implementation/src/torchvision/models/load_model.cs b/Implementation/src/torchvision/models/load_model.cs
@@ -22,7 +22,7 @@ static models()
 
                 internal static Dictionary<string, torch.Tensor> __load_model(string path, string fname, string url, bool print = true)
                 {
-                    if(File.Exists(path))
+                    if(File.Exists(path + fname))
                     {
                         return (Dictionary<string, torch.Tensor>)torch.load(path);
                     }
@@ -43,4 +43,4 @@ static models()
 
     }
 
-}
+}

Original file line number	Diff line number	Diff line change
`@@ -22,7 +22,7 @@ static models()`
`22`	`22`
`23`	`23`	`internal static Dictionary<string, torch.Tensor> __load_model(string path, string fname, string url, bool print = true)`
`24`	`24`	`{`
`25`		`- if(File.Exists(path))`
	`25`	`+ if(File.Exists(path + fname))`
`26`	`26`	`{`
`27`	`27`	`return (Dictionary<string, torch.Tensor>)torch.load(path);`
`28`	`28`	`}`
`@@ -43,4 +43,4 @@ static models()`
`43`	`43`
`44`	`44`	`}`
`45`	`45`
`46`		`-}`
	`46`	`+}`