Spaces:
Runtime error
Runtime error
File size: 283 Bytes
dc07399 |
1 2 3 4 5 6 7 8 9 10 11 12 |
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() |