Hugging Face
Models
Datasets
Spaces
Posts
Docs
Enterprise
Pricing
Log In
Sign Up
pitangent-ds
/
GPT2-genre-detection
like
0
Follow
Pitangent AI/ML Team
4
Text Classification
Transformers
TensorBoard
Safetensors
PyTorch
datadrivenscience/movie-genre-prediction
English
gpt2
GPT2ForSequenceClassification
Generated from Trainer
text-generation-inference
Inference Endpoints
License:
mit
Model card
Files
Files and versions
Metrics
Training metrics
Community
Train
Deploy
Use this model
main
GPT2-genre-detection
1 contributor
History:
4 commits
arnabdhar
Update README.md
5444752
11 months ago
runs
End of training
11 months ago
.gitattributes
Safe
1.52 kB
initial commit
11 months ago
README.md
Safe
2.06 kB
Update README.md
11 months ago
config.json
Safe
1.41 kB
End of training
11 months ago
merges.txt
Safe
456 kB
End of training
11 months ago
model.safetensors
Safe
498 MB
LFS
End of training
11 months ago
special_tokens_map.json
Safe
131 Bytes
End of training
11 months ago
tokenizer.json
Safe
2.11 MB
End of training
11 months ago
tokenizer_config.json
Safe
476 Bytes
End of training
11 months ago
training_args.bin
pickle
Detected Pickle imports (8)
"accelerate.utils.dataclasses.DistributedType"
,
"torch.device"
,
"transformers.training_args.OptimizerNames"
,
"transformers.trainer_utils.SchedulerType"
,
"transformers.training_args.TrainingArguments"
,
"transformers.trainer_utils.IntervalStrategy"
,
"accelerate.state.PartialState"
,
"transformers.trainer_utils.HubStrategy"
How to fix it?
4.73 kB
LFS
End of training
11 months ago
vocab.json
Safe
798 kB
End of training
11 months ago