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.2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 7fb086e6715bfb2c2857142d873e1c54a91e461d7f2ec11bc75ef01e5db1a464 |
|
MD5 | 64e876c67c6f6566428fed728ea787c0 |
|
BLAKE2b-256 | de99cf74a8c427cb487d9f8409373d6bb8e1ffce4e61af468fb780f3adc3ca08 |