Downloads · 30 days
5
11% of all-time downloads
roy-W/ired-reasoning
ired-reasoning is a machine learning model from roy-W. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for ired.
Frozen BART autoencoder with compression/reconstruction split.
Downloads · 30 days
5
11% of all-time downloads
All-time downloads
44
Public
Repo size
2.2 GB
Likes
0
Public
Click a slice to open those files.
.pt1.7 GB · 100%
From the Hugging Face model README
roy-W/ired-reasoningFrozen BART autoencoder with compression/reconstruction split.
convfacebook/bart-basefrom ired.model.autoencoder import FrozenBartAutoencoder
import torch
ae = FrozenBartAutoencoder(
model_name='facebook/bart-base',
k=384,
pool_type='conv',
d_ae=768,
train_decoder=True,
)
ckpt = torch.load('ae_checkpoint.pt', map_location='cpu')
ae.load_ae(ckpt['ae'])