36 lines
636 B
Go
36 lines
636 B
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
|
|
"git.andr3h3nriqu3s.com/andr3/gotch"
|
|
"git.andr3h3nriqu3s.com/andr3/gotch/pickle"
|
|
)
|
|
|
|
func main() {
|
|
// modelName := "vgg16"
|
|
// modelName := "mobilenet_v2"
|
|
// modelName := "resnet18"
|
|
// modelName := "alexnet"
|
|
// modelName := "squeezenet1_1"
|
|
// modelName := "inception_v3_google"
|
|
modelName := "efficientnet_b4"
|
|
|
|
url, ok := gotch.ModelUrls[modelName]
|
|
if !ok {
|
|
log.Fatalf("Unsupported model name %q\n", modelName)
|
|
}
|
|
modelFile, err := gotch.CachedPath(url)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
m, err := pickle.LoadModelInfo(modelFile)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
fmt.Println(m)
|
|
}
|