import tensorflow as tf

from data import DataHandler
from models import ACRegNet
import tensorflow as tf
from utils import get_random_batch, read_config_file, create_dir


RUN_IN_GPU = False


def train_acregnet_model(config):
    tf.reset_default_graph()
    tf_config = tf.ConfigProto()

    if RUN_IN_GPU:
        tf_config.gpu_options.allow_growth = True

    sess = tf.Session(config=tf_config)

    train_ims, _ = DataHandler.load_images(config['train_ims_file'])
    train_lbs, _ = DataHandler.load_labels(config['train_lbs_file'])
