geekfeiw / csi-net Goto Github PK

for (samples, labels) in tqdm(train_data_loader):
        samplesV = Variable(samples.cuda())
        labelsV = Variable(labels.cuda())
        predict_label = resnet(samplesV)
        fall_stand_preds = predict_label[:, 0]  # 取第一列
        correct_t += fall_stand_preds.eq(labelsV[:, 0].data.long()).sum()
print("Training accuracy:", (100*float(correct_t)/num_train_instances))

(model-venv) PS C:\Users\ashka\Desktop\summer 23\temple REU\wifi project\pretrained csi models\CSI-Net> python .\test.py
Traceback (most recent call last):
  File "C:\Users\ashka\Desktop\summer 23\temple REU\wifi project\pretrained csi models\CSI-Net\test.py", line 29, in <module>
    resnet = torch.load('weights/res18_Train100.0Test93.001.pkl')
  File "C:\Users\ashka\Desktop\summer 23\temple REU\wifi project\pretrained csi models\model-venv\lib\site-packages\torch\serialization.py", line 815, in load
    return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
  File "C:\Users\ashka\Desktop\summer 23\temple REU\wifi project\pretrained csi models\model-venv\lib\site-packages\torch\serialization.py", line 1043, in _legacy_load
    result = unpickler.load()
  File "C:\Users\ashka\Desktop\summer 23\temple REU\wifi project\pretrained csi models\model-venv\lib\site-packages\torch\serialization.py", line 848, in find_class
    return super().find_class(mod_name, name)
ModuleNotFoundError: No module named 'res_net_use_this'