summaryrefslogtreecommitdiff
path: root/research/knobloch/cmd/train/serialize.go
diff options
context:
space:
mode:
Diffstat (limited to 'research/knobloch/cmd/train/serialize.go')
-rw-r--r--research/knobloch/cmd/train/serialize.go4
1 files changed, 3 insertions, 1 deletions
diff --git a/research/knobloch/cmd/train/serialize.go b/research/knobloch/cmd/train/serialize.go
index 9fbf968..5536450 100644
--- a/research/knobloch/cmd/train/serialize.go
+++ b/research/knobloch/cmd/train/serialize.go
@@ -23,7 +23,9 @@ func serialize() {
paths = ps
}
- steps := []int{100512, 50256, 32768, 16384, 8192}
+ // steps := []int{100512, 50256, 32768, 16384, 8192}
+ // steps := []int{131072}
+ steps := []int{256}
outRoot := shelf.Abs("tokenizers")