Views
No views yet
1import torch
2from src.task1.models.lstm import LSTMDecryptionModel
3from src.task1.preprocessing.dataset import Vocabulary
4
5cipher_vocab = Vocabulary.load("cipher_vocab.json")
6plain_vocab = Vocabulary.load("plain_vocab.json")
7
8model = LSTMDecryptionModel(
9 cipher_vocab_size=cipher_vocab.size,
10 plain_vocab_size=plain_vocab.size,
11)
12ckpt = torch.load("model.pt", map_location="cpu")
13model.load_state_dict(ckpt["model_state"])
14model.eval()