Hello sir i have an other error when passing image to the model
Error is
Expected 4-dimensional input for 4-dimensional weight 16 3 3 3, but got 3-dimensional input of size [136, 136, 3] instead
and my code is
dataset = Dataset(csv_file='Dataset/loclization.csv', root_dir='Dataset/local/')

image_name,landmark=dataset[0]
print(landmark)
image = io.imread(image_name)
show_landmarks(image, landmark)
image=torch.from_numpy(image)
model(image)