1 2 3 4 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)