cannot convert float NaN to integer about deep-packet HOT 2 CLOSED

import torch
import pathlib
from ml.utils import load_cnn_model
from preprocessing import read_pcap, transform_packet
import torch.nn.functional as F
from utils import ID_TO_APP, ID_TO_TRAFFIC


if torch.cuda.is_available():
    print("GPU")

# model path
application = 'model/application_classification.cnn.model'
traffic = 'model/traffic_classification.cnn.model'

model = load_cnn_model(traffic, gpu=True)

print("Processing")

rows = []
batch_index = 0
for i, packet in enumerate(read_pcap(pathlib.Path("data/test/4.pcap"))):
    arr = transform_packet(packet)
    if arr is not None:
        y_pred = F.log_softmax(model(torch.Tensor(arr.todense().tolist()).cuda()), dim=1)
        y_hat = torch.argmax(y_pred, dim=1)
        # app_label = ID_TO_APP.get(int(y_hat))
        traffic_label = ID_TO_TRAFFIC.get(int(y_hat))
        print(traffic_label, round(y_pred.tolist()[0][int(y_hat)], 4))