from src.run import NLP_classification
import wandb
import yaml

config_file = 'configs/config.yaml'
config = yaml.load(open(config_file), Loader=yaml.FullLoader)

trainer = NLP_classification(**config['model'])

trainer.training(**config['train'], **config['wandb'])

wandb.finish()