#!/usr/bin/python
# -*- coding:utf-8 -*-
from .autoencoder_trainer import AutoEncoderTrainer
from .ldm_trainer import LDMTrainer

import utils.register as R


def create_trainer(config, model, train_loader, valid_loader):
    return R.construct(
        config['trainer'],
        model=model,
        train_loader=train_loader,
        valid_loader=valid_loader,
        save_config=config)