import torch
import numpy as numpy

class ExperimentBuilder(object):
    def __init__(self,data):
        self.data = data
        self.model = model


    def run_experiment():
        ## training and validate
        for epoch in range(100):
            train(self.data.get_train_batches)
            validate(self.data.get_val_batches)

        if args.save_model:
            torch.save(model.state_dict(),"model_test.pt")

    def train():
        

