Hermona/
my-first-model

Name: my-first-model
Availability: InStock

like

0

Views

No views yet

My First Model

This is my first model uploaded to Hugging Face!

Usage

python

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.

Hermona/my-first-model

My First Model

Usage

Hermona/
my-first-model