-
Notifications
You must be signed in to change notification settings - Fork 2
Description
Hello,
I am trying to train the data using "timesweeper train".
It keeps generating the following error. And I am a bit stuck here. Could you help me?
2024-04-09 11:59:39.997844: I external/local_tsl/tsl/cuda/cudart_stub.cc:32] Could not find cuda drivers on your machine, GPU will not be used.
2024-04-09 11:59:40.001892: I external/local_tsl/tsl/cuda/cudart_stub.cc:32] Could not find cuda drivers on your machine, GPU will not be used.
2024-04-09 11:59:40.051582: I tensorflow/core/platform/cpu_feature_guard.cc:210] This TensorFlow binary is optimized to use available CPU instructions in performance-critical operations.
To enable the following instructions: AVX2 FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags.
2024-04-09 11:59:40.943383: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Could not find TensorRT
INFO:nets:Random seed is: 104644
Traceback (most recent call last):
File "/home/kiwoong/miniconda3/envs/blinx/bin/timesweeper", line 8, in
sys.exit(ts_main())
^^^^^^^^^
File "/home/kiwoong/miniconda3/envs/blinx/lib/python3.12/site-packages/timesweeper/cli.py", line 472, in ts_main
train_nets.main(ua)
File "/home/kiwoong/miniconda3/envs/blinx/lib/python3.12/site-packages/timesweeper/train_nets.py", line 461, in main
experiment_name = yaml_data["experiment name"]
~~~~~~~~~^^^^^^^^^^^^^^^^^^^
KeyError: 'experiment name'
Thank you very much.
Best,
Kiwoong