jy4140 commited on
Commit
e905c9a
1 Parent(s): e123368

Revert "Training in progress, epoch 1"

Browse files

This reverts commit f73bb40168ca71ca8ec163ad991d992edca6672f.

Files changed (3) hide show
  1. config.json +35 -0
  2. model.safetensors +4 -0
  3. training_args.bin +3 -0
config.json ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "distilbert/distilbert-base-uncased",
3
+ "activation": "gelu",
4
+ "architectures": [
5
+ "DistilBertForSequenceClassification"
6
+ ],
7
+ "attention_dropout": 0.1,
8
+ "dim": 768,
9
+ "dropout": 0.1,
10
+ "hidden_dim": 3072,
11
+ "id2label": {
12
+ "0": "Safe",
13
+ "1": "Unsure",
14
+ "2": "Unsafe"
15
+ },
16
+ "initializer_range": 0.02,
17
+ "label2id": {
18
+ "Safe": 0,
19
+ "Unsafe": 2,
20
+ "Unsure": 1
21
+ },
22
+ "max_position_embeddings": 512,
23
+ "model_type": "distilbert",
24
+ "n_heads": 12,
25
+ "n_layers": 6,
26
+ "pad_token_id": 0,
27
+ "problem_type": "single_label_classification",
28
+ "qa_dropout": 0.1,
29
+ "seq_classif_dropout": 0.2,
30
+ "sinusoidal_pos_embds": false,
31
+ "tie_weights_": true,
32
+ "torch_dtype": "float32",
33
+ "transformers_version": "4.40.0",
34
+ "vocab_size": 30522
35
+ }
model.safetensors CHANGED
@@ -1,3 +1,7 @@
1
  version https://git-lfs.github.com/spec/v1
 
2
  oid sha256:f3399e1b47ec179d0e48d1957798b8790995113217ad665df68f6173d5163e5f
 
 
 
3
  size 267835644
 
1
  version https://git-lfs.github.com/spec/v1
2
+ <<<<<<< HEAD
3
  oid sha256:f3399e1b47ec179d0e48d1957798b8790995113217ad665df68f6173d5163e5f
4
+ =======
5
+ oid sha256:1163f7a5b0c53fc893565f52a29373b61f250f7d45cbd9348b209ca61feace1b
6
+ >>>>>>> parent of f73bb40 (Training in progress, epoch 1)
7
  size 267835644
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e16d492950712062e2492baf05548ba10ba85b338db2897c3cc5868fb5f7540d
3
+ size 4984