Traceback (most recent call last):
File "/root/Mortal/mortal/train.py", line 424, in <module>
main()
File "/root/Mortal/mortal/train.py", line 402, in main
train()
File "/root/Mortal/mortal/train.py", line 383, in train
train_epoch()
File "/root/Mortal/mortal/train.py", line 264, in train_epoch
writer.add_histogram('q_predicted', all_q_1d, steps)
File "/root/miniconda3/envs/mortal/lib/python3.10/site-packages/torch/utils/tensorboard/writer.py", line 485, in add_histogram
histogram(tag, values, bins, max_bins=max_bins), global_step, walltime
File "/root/miniconda3/envs/mortal/lib/python3.10/site-packages/torch/utils/tensorboard/summary.py", line 358, in histogram
hist = make_histogram(values.astype(float), bins, max_bins)
File "/root/miniconda3/envs/mortal/lib/python3.10/site-packages/torch/utils/tensorboard/summary.py", line 386, in make_histogram
cum_counts = np.cumsum(np.greater(counts, 0, dtype=np.int32))
TypeError: No loop matching the specified signature and casting was found for ufunc greater
numpy 1.24+ 对数据类型更加严格导致train报错
降级1.23临时解决
pip install numpy==1.23