CoST
CoST copied to clipboard
salesforce
einsum() operands do not broadcast with remapped shapes
Hi, Gerald. I am trying to running your code on my own dataset. But I got some problems here:
RuntimeError: einsum() operands do not broadcast with remapped shapes [original->remapped]: [32, 37, 320]->[32, 37, 1, 320] [101, 320, 160]->[1, 101, 160, 320]
Hey, thanks for your interest in our work, could you provide more information on this error (the full error log) and the dimensions of the data you are working with? I can't really tell whats wrong from just this error message.
I met the same problem. ` import torch
from torch import nn
import torchinfo
from torchinfo import summary
net = CoSTEncoder( input_dims=12, output_dims=3, kernels=[1, 2, 4, 8, 16, 32, 64, 128], length=3000, hidden_dims=64, depth=10, )
summary(net,input_size=(1,121,12),col_names=["kernel_size","output_size","num_params","mult_adds"]) ` And I get the error einsum(): operands do not broadcast with remapped shapes [original->remapped]: [1, 61, 3]->[1, 61, 1, 3] [1501, 3, 1]->[1, 1501, 1, 3]
I figure out the problem, the length in CoSTEncoder must match your dataset. For example, my length is 121, then change it in the Enocder part. ` import torch
from torch import nn
import torchinfo
from torchinfo import summary
net = CoSTEncoder( input_dims=12, output_dims=3, kernels=[1, 2, 4, 8, 16, 32, 64, 128], length=121, hidden_dims=64, depth=10, ) ` summary(net,input_size=(1,121,12),col_names=["kernel_size","output_size","num_params","mult_adds"])
My dataset is similar to yours, with a total of 8 columns and 1681 rows, and the runtime reports such an error
Traceback (most recent call last):
File "train.py", line 109, in
