When i load a model i am seeing this error?

Your code is neither properly formatted nor is it executable since you have a data dependency.
In any case, the error is raised since you are mixing up input arguments in the model creation.
At first you are using:

model_x = tinyvgg1(input=3, hidden=76, output=2).to(device)

and later:

model = tinyvgg1(3,76,2).to(device)

which corresponds to tinyvgg1(hidden=3, input=76, output=2).

1 Like