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()