26 lines
593 B
Go
26 lines
593 B
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
|
|
"git.andr3h3nriqu3s.com/andr3/gotch/ts"
|
|
)
|
|
|
|
func main() {
|
|
// NOTE. Python script to save model to .npz can be found at https://github.com/sugarme/pytorch-pretrained/bert/bert-base-uncased-to-npz.py
|
|
filepath := "../../data/convert-model/bert/model.npz"
|
|
|
|
namedTensors, err := ts.ReadNpz(filepath)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
fmt.Printf("Num of named tensor: %v\n", len(namedTensors))
|
|
outputFile := "/home/sugarme/projects/transformer/data/bert/model.gt"
|
|
err = ts.SaveMultiNew(namedTensors, outputFile)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|