solarlee / FewSAR

5 stars 0 forks source link

使用ATLNet测试结果全部为nan #1

Open Wang-jun-yu opened 9 months ago

Wang-jun-yu commented 9 months ago

2023-11-14 21:57:27,362 [INFO] core.test: Test Accuracy: nan h: nan 2023-11-14 21:57:27,363 [INFO] core.test: ============ Testing on the test set ============ 2023-11-14 21:57:27,421 [INFO] core.test: Test Accuracy: nan h: nan 2023-11-14 21:57:27,422 [INFO] core.test: ============ Testing on the test set ============ 2023-11-14 21:57:27,487 [INFO] core.test: Test Accuracy: nan h: nan 2023-11-14 21:57:27,487 [INFO] core.test: ============ Testing on the test set ============ 2023-11-14 21:57:27,541 [INFO] core.test: Test Accuracy: nan h: nan 2023-11-14 21:57:27,542 [INFO] core.test: ============ Testing on the test set ============ 2023-11-14 21:57:27,595 [INFO] core.test: Test Accuracy: nan h: nan 2023-11-14 21:57:27,595 [INFO] core.test: Aver Accuracy: nan Aver h: nan 2023-11-14 21:57:27,596 [INFO] core.test: ............Testing is end............

Process finished with exit code 0

使用给出的示例"./config/atl_net.yaml"进行训练后,运行run_test.py,结果全部为nan。

run_test.py文件如下: sys.dont_write_bytecode = True

PATH = r"E:\study\few-shot\FewSAR-main\dataset\MSTAR-ATLNet-Conv64F-5-1-Nov-14-2023-21-36-27"

PATH = r"E:\study\few-shot\FewSAR-main\dataset\MSTAR-ATLNet-Conv64F-5-1-Nov-14-2023-21-36-27"

VAR_DICT = { "test_epoch" : 5, "device_ids" : "2", "n_gpu" : 1, "test_episode": 600, "episode_size": 1, "test_way" : 6, }

if name == "main": config = Config(os.path.join(PATH, "config.yaml"), VAR_DICT).get_config_dict() test = Test(config, PATH) test.test_loop()

XDS0rryMaker commented 9 months ago

我也是,有什么解决方法吗?

Fvpder commented 3 months ago

test_way的样本种类改成和训练时相同的,默认应该是5