pg2455 / KAN-Tutorial

Understanding Kolmogorov-Arnold Networks: A Tutorial Series on KAN using Toy Examples
MIT License
167 stars 13 forks source link

Tutorial #4 breaks if KAN shape is changed #2

Closed celrm closed 4 months ago

celrm commented 4 months ago

RuntimeError: einsum(): operands do not broadcast with remapped shapes [original->remapped]: [4, 2, 12]->[1, 4, 2, 12] [1280, 2, 12]->[1280, 2, 1, 12] -- nvm the first value is not a variable it's just the input size