Hermona/my-first-model

My First Model

This is my first model uploaded to Hugging Face!

Usage

import torch

class TinyModel(torch.nn.Module):
    def __init__(self):
        super().__init__()
        self.layer = torch.nn.Linear(10, 2)
    
model = TinyModel()
model.load_state_dict(torch.load('pytorch_model.bin'))
created by Hermona.