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()