Spaces:
Runtime error
Runtime error
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() |