metadata
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
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