File “/voyager-volume/code_1_test/code_1_test/code_1_test/code_1_test/code_1_test/architect.py”, line 158, in _construct_model_from_theta
params[k] = theta[offset: offset+v_length].view(v.size())
RuntimeError: shape ‘[48]’ is invalid for input of size 0
@ptrblck ,could you tell why I am facing this issue