search

yrmo / cudagrad

CUDA C++ strided float tensor automatic differentiation engine with Python bindings
MIT License
0 stars 0 forks source link

Show matrix dimensions when in invalid dimensions error #57

Open yrmo opened 4 months ago

yrmo commented 4 months ago 
$ python mnist.py 
Traceback (most recent call last):
  File "/home/ryan/cudagrad/examples/mnist.py", line 61, in <module>
    a = accuracy()
  File "/home/ryan/cudagrad/examples/mnist.py", line 48, in accuracy
    outputs.append(int(model(train_images[i]).item()))
  File "/home/ryan/cudagrad/examples/mnist.py", line 39, in __call__
    self.w1 @ Tensor.sigmoid((self.w0 @ x + self.b0)) + self.b1
RuntimeError: Invalid dimensions for matrix multiplication.