Views
No views yet
1import torch
2from spiking_neural_net import SimpleSpikingNeuralNetwork
3
4# 加載模型
5model = SimpleSpikingNeuralNetwork(
6 input_size=2312,
7 hidden_size=100,
8 output_size=10,
9 time_window=20
10)
11model.load_state_dict(torch.load("model.pt"))
12
13# 使用模型進行預測
14model.eval()
15with torch.no_grad():
16 predictions = model(input_data)