summaryrefslogtreecommitdiff
path: root/research
diff options
context:
space:
mode:
Diffstat (limited to 'research')
-rw-r--r--research/knobloch/cmd/train/main.go7
1 files changed, 7 insertions, 0 deletions
diff --git a/research/knobloch/cmd/train/main.go b/research/knobloch/cmd/train/main.go
index a946ce4..d2f49f2 100644
--- a/research/knobloch/cmd/train/main.go
+++ b/research/knobloch/cmd/train/main.go
@@ -10,14 +10,21 @@ import (
"github.com/jonasknobloch/mbpe"
"go.jknobloc.com/x/dataset"
+ "go.jknobloc.com/x/profile"
"go.jknobloc.com/x/shelf"
"go.jknobloc.com/x/tokenizer/bpe"
"go.jknobloc.com/x/tokenizer/bpe/split"
)
func main() {
+ stop := profile.CPU()
+
// train()
// serialize()
+
+ profile.Mem()
+
+ stop()
}
func train() {