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)