GuakGuak's picture
add
dc07399
raw
history blame contribute delete
No virus
283 Bytes
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()