dataclass + nn.Module
Project description
DataclassModule
dataclass + nn.Module
Usage
from dataclassmodule import DataclassModule
class ExampleModule(DataclassModule):
x: int
def __post_init__(self):
self.layer = nn.Linear(2, 2)
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Close
Hashes for torch_dataclasses-0.1.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 2e1289bd632cfce7aaa259ff3f8c0e6a6150bb44591c55b287178b2021629913 |
|
MD5 | 272e428d188621f43f1fa0997d3a1c40 |
|
BLAKE2b-256 | af60783a174e951fc0f8bd491bcdd43af0b064e218b8856d5e33ac562d6e862d |