NielsRogge/transformers

Pull Requests Commits

dataclass doesn't reorder optional args, remove default value for device.

mfuntowicz committed 5 years ago

7f579cb3

Specify device in DataCollator in order to place tensors directly on the right device and avoid moving data around afterwards.

mfuntowicz committed 5 years ago

66677262

Unused Union should not be imported

LysandreJik committed 5 years ago

a0865277

TPU hangs when saving optimizer/scheduler (#4467)

LysandreJik committed 5 years ago

Verified 9d2ce253

Adds predict stage for glue tasks, and generate result files which can be submitted to gluebenchmark.com (#4463)

stdcoutzyx committed 5 years ago

Verified 49296533

[examples] fix no grad in second pruning in run_bertology (#4479)

Tobias Lee committed 5 years ago

Verified 271bedb4

[ci] Close #4481

julien-c committed 5 years ago

865d4d59

Update test_trainer_distributed.py

julien-c committed 5 years ago

a3af8e86

:rotating_light: Remove warning of deprecation (#4477)

astariul committed 5 years ago

Verified eacea530

Better None gradients handling in TF Trainer (#4469)

jplu committed 5 years ago

Verified fa2fbed3

Correct TF formatting to exclude LayerNorms from weight decay (#4448)

oliverastrand committed 5 years ago

Verified e708bb75

pass on tokenizer to pipeline (#4489)

RensDimmendaal committed 5 years ago

Verified 49c06132

Add Fine-tune DialoGPT on new datasets notebook (#4473)

Nathan Cooper committed 5 years ago

Verified cacb654c

Adjust german bert model card, add new model card (#4488)

Timoeller committed 5 years ago

Verified 30a09f38

Fix slow gpu tests lysandre (#4487)

LysandreJik committed 5 years ago

Verified 14cb5b35

Create README.md (#4482)

mrm8488 committed 5 years ago

Verified 6dc52c78

Model card for RuPERTa-base fine-tuned for NER (#4466)

mrm8488 committed 5 years ago

Verified ed5456da

Model card for Tereveni-AI/gpt2-124M-uk-fiction (#4470)

obsh committed 5 years ago

Verified c76450e2

add BERT trained from review corpus. (#4405)

howardhsu committed 5 years ago

Verified 9907dc52

[MarianTokenizer] implement save_vocabulary and other common methods (#4389)

sshleifer committed 5 years ago

Verified efbc1c5a

[gpu slow tests] fix mbart-large-enro gpu tests (#4472)

sshleifer committed 5 years ago

Verified 956c4c4e

[Longformer] Docs and clean API (#4464)

patrickvonplaten committed 5 years ago

Verified 48c3a70b

[Tests, GPU, SLOW] fix a bunch of GPU hardcoded tests in Pytorch (#4468)

patrickvonplaten committed 5 years ago

Verified aa925a52

add T5 fine-tuning notebook [Community notebooks] (#4462)

patil-suraj committed 5 years ago

Verified 5856999a

[cleanup] test_tokenization_common.py (#4390)

sshleifer committed 5 years ago

Verified 07dd7c2f

Longformer (#4352)

ibeltagy committed 5 years ago

Verified 8f1d0471

Refactored the README.md file (#4427)

girishponkiya committed 5 years ago

Verified 31eedff5

Map optimizer to correct device after loading from checkpoint. (#4403)

shaoyent committed 5 years ago

Verified 384f0eb2

[Trainer] move model to device before setting optimizer (#4450)

julien-c committed 5 years ago

Verified bf14ef75

Distributed eval: SequentialDistributedSampler + gather all results (#4243)

julien-c committed 5 years ago

Verified 5e7fe8b5

Older