QARAC / qarac /models /QaracTrainerModel.py

Commit History

Diagnostics
1a9032d

PeteBleackley commited on

Diagnostics
3fc3a1b

PeteBleackley commited on

Diagnostics
7758fd9

PeteBleackley commited on

Correct dimension of consistency cosine
14d83dc

PeteBleackley commited on

Using torch.nn.CosineSimilarity to simplify code
798488e

PeteBleackley commited on

Use keepdim option when normalising vectors
738b546

PeteBleackley commited on

Make EPSILON a tensor
cf5f935

PeteBleackley commited on

torch.maximum, not torch.max
98ad67d

PeteBleackley commited on

PyTorch implementation of HugggingFace PreTrainedModel class does not allow direct setting of base_model. Rejig constructors accordingly
519dfd1

PeteBleackley commited on

Removed superfluous ()
518e821

PeteBleackley commited on

Converted QaracTrainerModel to use PyTorch
56e5680

PeteBleackley commited on

Removed extraneous self
ae31ae3

PeteBleackley commited on

Superclass __init__
8626c30

PeteBleackley commited on

Fix typos
d0cb26f

PeteBleackley commited on

Fix typos
e563d3d

PeteBleackley commited on

Fix imports
23a97a2

PeteBleackley commited on

Components for managing training corpora
4f8366b

PeteBleackley commited on

More work on models
f16a715

PeteBleackley commited on

Encoder, Decoder and Trainer models (assuming RoBERTa base models)
f9c0522

PeteBleackley commited on