From 49375838a74bc8519f7ac9ec9a983316888bdacc Mon Sep 17 00:00:00 2001
From: Marcin Chrzanowski <m@m-chrzan.xyz>
Date: Sun, 23 May 2021 13:20:53 +0200
Subject: Add train loop skeleton

---
 train/__main__.py | 5 +++++
 1 file changed, 5 insertions(+)
 create mode 100644 train/__main__.py

(limited to 'train/__main__.py')

diff --git a/train/__main__.py b/train/__main__.py
new file mode 100644
index 0000000..ec544f7
--- /dev/null
+++ b/train/__main__.py
@@ -0,0 +1,5 @@
+from train.train import train_model
+from model.encoder import EncoderModel
+
+model = EncoderModel(16, 64, 128, 10, 4, 4)
+train_model(model, 0.1, 100, 10)
-- 
cgit v1.2.3