File size: 764 Bytes
78b6fa8 96bcc02 78b6fa8 96bcc02 78b6fa8 96bcc02 78b6fa8 96bcc02 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
---
language: en
license: mit
tags:
- simple_linear_model
library_name: transformers
---
# Custom Linear Model
This is a simple custom model with a single linear layer.
It does not have practical usages, just used for testing.
## Model Details
- Input size: 15000
- Output size: 15000
- Architecture: Single linear layer
## Usage
To start with clone the repo
git clone https://huggingface.co/sushilks/simple_linear_model
```python
from simple_linear_model.model import *
from transformers import AutoConfig, AutoModel
from transformers import AutoModel
model = AutoModel.from_pretrained("sushilks/simple_linear_model_15k")
# Forward pass
import torch
x = torch.randn(1, 15000) # Example input
output = model(x)
print(output.shape) # shoudl be 1,15000
```
|