ChebAutoencoder / eval.py
lara roth
Update AI model
faeee34
Raw
History Blame Contribute Delete
220 Bytes
import torch
from AutoencoderCheb import AutoEncoderModel
model = AutoEncoderModel(cuda_true=True, batch_size=32)
model.load_state_dict(torch.load("ChebAutoencoder/gae_25_08_2025.pt"))
print("model", model)
model.eval()