Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3tokenizer = AutoTokenizer.from_pretrained("path_to_model")
4model = AutoModelForCausalLM.from_pretrained("path_to_model")
5
6inputs = tokenizer("Your input text", return_tensors="pt")
7outputs = model(**inputs)
8
9[More Information Needed]
10
11## Training Details
12
13### Training Data
14
15The dataset can be downloaded from https://shorturl.at/gQKY6.
16
17### Training Procedure
18
19<!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
20
21#### Preprocessing [optional]
22
23
24
25#### Training Hyperparameters
26
27Training Hyperparameters
28Training regime: bf16 mixed precision
29Learning rate: 2e-4
30Batch size: 1
31Gradient accumulation steps: 4
32Warmup steps: 2
33Max steps: 20
34Optimizer: paged_adamw_8bit
35Evaluation strategy: steps
36Evaluation steps: 1
37
38#### Speeds, Sizes, Times [optional]
39
40<!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
41
42[More Information Needed]
43
44## Evaluation
45
46<!-- This section describes the evaluation protocols and provides the results. -->
47
48### Testing Data, Factors & Metrics
49
50#### Testing Data
51
52The model was evaluated on the SciNLI test set, containing sentence pairs from scientific papers.
53
54#### Factors
55
56[More Information Needed]
57
58#### Metrics
59
60<!-- These are the evaluation metrics being used, ideally with a description of why. -->
61
62[More Information Needed]
63
64### Results
65
66[More Information Needed]
67
68#### Summary
69
70
71
72## Model Examination [optional]
73
74<!-- Relevant interpretability work for the model goes here -->
75
76[More Information Needed]
77
78## Environmental Impact
79
80<!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
81
82Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
83
84- **Hardware Type:** [More Information Needed]
85- **Hours used:** [More Information Needed]
86- **Cloud Provider:** [More Information Needed]
87- **Compute Region:** [More Information Needed]
88- **Carbon Emitted:** [More Information Needed]
89
90## Technical Specifications [optional]
91
92### Model Architecture and Objective
93
94[More Information Needed]
95
96### Compute Infrastructure
97
98[More Information Needed]
99
100#### Hardware
101
102[More Information Needed]
103
104#### Software
105
106[More Information Needed]
107
108## Citation [optional]
109
110<!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
111
112**BibTeX:**
113
114[More Information Needed]
115
116**APA:**
117
118[More Information Needed]
119
120## Glossary [optional]
121
122<!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
123
124[More Information Needed]
125
126## More Information [optional]
127
128[More Information Needed]
129
130## Model Card Authors [optional]
131
132[More Information Needed]
133
134## Model Card Contact
135
136[More Information Needed]
137