VisualJoyce/ChengyuBERT

error when training chengyubert-twostage

Closed this issue · 3 comments

`
[1,0]:
[1,0]:Traceback (most recent call last):
[1,0]: File "train_official.py", line 468, in
[1,0]: main(args)
[1,0]: File "train_official.py", line 317, in main
[1,0]: best_ckpt = train(model, dataloaders, opts)
[1,0]: File "train_official.py", line 145, in train
[1,0]: opts, global_step)
[1,0]: File "train_official.py", line 235, in evaluation
[1,0]: log.update(validate(opts, model, loader, split, global_step))
[1,0]: File "/opt/conda/lib/python3.7/site-packages/torch/autograd/grad_mode.py", line 15, in decorate_context
[1,0]: return func(*args, **kwargs)
[1,0]: File "train_official.py", line 176, in validate
[1,0]: logits, over_logits, cond_logits = model(**batch, targets=None, compute_loss=False)
[1,0]:ValueError: not enough values to unpack (expected 3, got 2)

`

chengyubert-twostage is not released yet because the paper is still under review.

oh...Thanks a lot

Two stage is released!