pytorch-fitmodule
pytorch-fitmodule copied to clipboard
fix the "IndexError: invalid index of a 0-dim tensor. Use tensor.item…
…() to convert a 0-dim tensor to a Python", that is a pytorch version issue