Downloads · 30 days
5
29% of all-time downloads
sushilks/simple_linear_model_1k
simple_linear_model_1k is a machine learning model from sushilks. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as mit.
This is a simple custom model with a single linear layer. It does not have practical usages, just used for testing.
Downloads · 30 days
5
29% of all-time downloads
All-time downloads
17
Public
Parameters
1M
8 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors4 MB · 100%
From the Hugging Face model README
This is a simple custom model with a single linear layer. It does not have practical usages, just used for testing.
To start with clone the repo git clone https://huggingface.co/sushilks/simple_linear_model
from simple_linear_model.model import *
from transformers import AutoConfig, AutoModel
from transformers import AutoModel
model = AutoModel.from_pretrained("sushilks/simple_linear_model_1k")
# Forward pass
import torch
x = torch.randn(1, 1000) # Example input
output = model(x)
print(output.shape) # shoudl be 1,1000