model = CifarVggnet(*args, **kwargs)
model.load_state_dict(torch.load(ckpt_path))
print(model)
I did that , but during loading I get error:
File “Checking.py”, line 17, in 
model = CifarVggnet(*args, **kwargs)
NameError: name ‘CifarVggnet’ is not defined