This repository was archived by the owner on Feb 1, 2025. It is now read-only.
RuntimeError: invalid argument 2: k not in range for dimension. #52
Description
Hi,
I followed the requirements to install the environment.
But when I run the scripts/run_experiments.py, there is a bug in __max_probs_values_indices:
RuntimeError: invalid argument 2: k not in range for dimension at /pytorch/aten/src/TH/generic/THTensorMoreMath.cpp:1190
I checked the log_prob is 28 but k is 10000, which exceeds.