NielsRogge/transformers

Pull Requests Commits

WIP closure instead of subclass

rlouf committed 6 years ago

2c6ebf1a

unpack summaries

rlouf committed 6 years ago

a8cf4a4f

pass pad token to mask builder

rlouf committed 6 years ago

27a281cb

rlouf committed 6 years ago

717c9d8c

specify tensor device in beam search

rlouf committed 6 years ago

ca9b6f32

tweaks to the BeamSearch API

rlouf committed 6 years ago

dae4df95

Share pre-trained embedding with the randomly initialized decoder

rlouf committed 6 years ago

15a7edd4

rlouf committed 6 years ago

5c202cd8

add pyrouge requirement

rlouf committed 6 years ago

5570486c

create new generate module

rlouf committed 6 years ago

f33fb25c

finalize BS + tests

rlouf committed 6 years ago

bac3e63f

update masks during beam search

rlouf committed 6 years ago

533228af

rlouf committed 6 years ago

c2708557

use BeamSearch + evaluate on ROUGE

rlouf committed 6 years ago

5a06910a

test and fix removal of repeating 3-grams

rlouf committed 6 years ago

4eb87588

test min_length & max_length constraints in Beam Search

rlouf committed 6 years ago

1f52a892

test on minimum length and beam termination pass

rlouf committed 6 years ago

cabcf152

WIP - <3 <3 Beam Search <3 <3

rlouf committed 6 years ago

9983c1d0

load the pretrained weights for encoder-decoder

rlouf committed 6 years ago

cd478c35

allow to save checkpoint on keyboard interrupt

rlouf committed 6 years ago

aa24121e

update function to add special tokens

rlouf committed 6 years ago

9b30045c

Add RoBERTa-based GPT-2 Output Detector from OpenAI

julien-c committed 6 years ago

1c542df7

Fix other PyTorch models

julien-c committed 6 years ago

2f3a4210

julien-c committed 6 years ago

d5319793

[tests] Flag to test on cuda

julien-c committed 6 years ago

27e015bd

[tests] get rid of warning

julien-c committed 6 years ago

13d9135f

[run_tf_glue] Add comment for context

julien-c committed 6 years ago

f88c104d

julien-c committed 6 years ago

30968d70

Updating docblocks in optimizers.py

Dom Hudson committed 6 years ago

de890ae6

LysandreJik committed 6 years ago

d7d36181

Older