castorini / hedwig

PyTorch deep learning models for document classification

castorini/hedwig Issues

UnpicklingError with different BERT models
Updated 2 years ago
BUG report
Updated 3 years ago
Paper of HBERT
Updated 3 years ago1
BertForSequenceClassification prediction.
Closed 5 years ago2
[DocBERT] Can DocBERT bin any length document without truncaiton?
Updated 4 years ago1
Scaled embedded dropout mask
Closed 4 years ago1
labels of the data
Updated 4 years ago
[Question] What are the "robust" datasets?
Updated 4 years ago
Getting impossible predicted labels (all zeroes) from custom data
Updated 4 years ago
Getting AttributeError: Can't get attribute 'gelu' on <module 'transformers.modeling_bert' in "/hedwig/models/bert/__main__.py", l
Closed 4 years ago
Killed message on trying models.reg_lstm
Updated 4 years ago1
does DocBERT freeze all BERT layers and add a fully connected layer on the top for classification?
Updated 4 years ago
Custom dataset error
Updated 4 years ago4
Where I can get the KD model
Updated 4 years ago
Bert Inference Issue
Closed 4 years ago1
Fine-tuning BERT for a new dataset
Updated 4 years ago
Alternate hosting for hedwig-data
Closed 4 years ago8
DocBERT
Closed 4 years ago3
Pretrained weights via model zoo
Updated 4 years ago2
Migrate from pytorch-pretrained-bert to transformers
Closed 4 years ago1
Calling BertTokenizer.from_pretrained() with the path to a single file or url is deprecated
Updated 5 years ago1
Reg-LSTM
Closed 5 years ago1
fasttext
Updated 5 years ago1
DocRoBERTa
Updated 5 years ago
setup
Closed 5 years ago
IMDB dataset
Updated 5 years ago
Memory error with Reuters and models.reg_lstm
Closed 5 years ago2
How to feed the document data into DocBert
Updated 5 years ago9
Early stopping criteria
Closed 5 years ago2
Correct save location for trained model
Updated 5 years ago1
fixed-size vector for each large document
Closed 5 years ago2
Testing a trained model throws error
Updated 5 years ago1
Replace model caching mechanism
Closed 5 years ago2
AttributeError: 'Tensor' object has no attribute 'uniform'
Updated 5 years ago1
Number of test and dev lables are not similar to original data
Updated 5 years ago
Failed saving the best model according to hedwig/models/bert/README.md
Closed 5 years ago6
Remove tensorboardX logging
Closed 5 years ago1
Is there a docker container?
Closed 5 years ago1
How do I use this with another dataset?
Closed 5 years ago1
self.model.train() should be outside the train-epoch loop
Closed 6 years ago
Reproducing Setup and README steps
Closed 6 years ago3
DocBERT model weights
Closed 6 years ago2
Add type hints
Updated 6 years ago
Use NLTK sent_tokenize and word_tokenize
Updated 6 years ago