diff options
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) |