报错解决——TypeError: No loop matching the specified signature and casting was found for ufunc greater

完整报错：

Traceback (most recent call last):
  File "train.py", line 212, in <module>
    train(epoch)
  File "train.py", line 71, in train
    writer.add_histogram("{}/{}".format(layer, attr), param, epoch)
  File "/home/amax/anaconda3/envs/test/lib/python3.8/site-packages/torch/utils/tensorboard/writer.py", line 441, in add_histogram
    histogram(tag, values, bins, max_bins=max_bins), global_step, walltime)
  File "/home/amax/anaconda3/envs/test/lib/python3.8/site-packages/torch/utils/tensorboard/summary.py", line 320, in histogram
    hist = make_histogram(values.astype(float), bins, max_bins)
  File "/home/amax/anaconda3/envs/test/lib/python3.8/site-packages/torch/utils/tensorboard/summary.py", line 344, 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

解决办法：

将

cum_counts = np.cumsum(np.greater(counts, 0, dtype=np.int32))

换成

cum_counts = np.cumsum(np.greater(counts, 0))