-
Notifications
You must be signed in to change notification settings - Fork 0
Description
Hi, When I run this code, this problem appears. Do I need to generate this TXT file or download it myself?If I need to generate it, what should I do? Thank you!
(tf-gpu) ubuntu@ubuntu:/mnt/be5e5f17-b8b5-496f-851a-c061a6576502/project/SL/Prototype-based-Memory-Network-main$ python main_pmnet.py --data_config='ucm2mai' --backbone='resnet50' --pretrain_weight_path='./weights/ucm2mai/pm-resnet50.h5' --weight_path='./weights/ucm2mai/resnet50-ucm.h5' --ep 100 --lr 5e-4 --evaluate 0
tensorflow version: 2.4.1
2021-05-10 15:14:45.593635: I tensorflow/core/platform/cpu_feature_guard.cc:142] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations: SSE4.1 SSE4.2 AVX AVX2 FMA
To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
2021-05-10 15:14:45.595281: I tensorflow/core/common_runtime/process_util.cc:146] Creating new thread pool with default inter op setting: 2. Tune using inter_op_parallelism_threads for best performance.
############## gpu memory is fixed to 0.3 on 0 #################
==================== Loading data ====================
Traceback (most recent call last):
File "main_pmnet.py", line 68, in
main()
File "main_pmnet.py", line 32, in main
X_tr, y_tr, X_test, y_test = load_data(args.data_config, args.patch_size, args.evaluate)
File "/mnt/be5e5f17-b8b5-496f-851a-c061a6576502/project/SL/Prototype-based-Memory-Network-main/utils/mai.py", line 91, in load_data
with open(config_test) as f:
FileNotFoundError: [Errno 2] No such file or directory: 'data/MAI_dataset/configs/ucm2mai_test.txt'