diff options
| author | Marcin Chrzanowski <m@m-chrzan.xyz> | 2021-05-23 13:20:53 +0200 |
|---|---|---|
| committer | Marcin Chrzanowski <m@m-chrzan.xyz> | 2021-05-23 13:20:53 +0200 |
| commit | 49375838a74bc8519f7ac9ec9a983316888bdacc (patch) | |
| tree | e82866674220ea5cfe96d4fefa3bd89d264653e6 /train/__main__.py | |
| parent | 9563a14c793668024543f931d80e7ca8b31fa577 (diff) | |
Add train loop skeleton
Diffstat (limited to 'train/__main__.py')
| -rw-r--r-- | train/__main__.py | 5 |
1 files changed, 5 insertions, 0 deletions
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) |