diff options
Diffstat (limited to 'llm/cmd/eval/main.go')
| -rw-r--r-- | llm/cmd/eval/main.go | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/llm/cmd/eval/main.go b/llm/cmd/eval/main.go index 78a865e..43752d9 100644 --- a/llm/cmd/eval/main.go +++ b/llm/cmd/eval/main.go @@ -9,8 +9,16 @@ import ( ) func main() { + if err := gpt2.InitializeEnvironment(); err != nil { + log.Fatal(err) + } + perplexity() // logprobs() + + if err := gpt2.DestroyEnvironment(); err != nil { + log.Fatal(err) + } } func data() *dataset.ParquetReader { |