nasnet-pytorch

nasnet-pytorch copied to clipboard

I defined a new function `def build_and_load_model(path, nas_type): model = NASNetALarge(1001) if nas_type == 'large' else NASNetAMobile(1001) filename_model = os.path.join(path, 'pytorch', nas_type == 'large' and 'nasnet_a_large.pth' or 'nasnet_a_mobile.pth') state_dict =...

alixiu

error while loading pretrained weights

2

I get this error while loading the weights: ``` python RuntimeError Traceback (most recent call last) in () ----> 1 model=model.load_state_dict(weights) /opt/conda/lib/python3.6/site-packages/torch/nn/modules/module.py in load_state_dict(self, state_dict, strict) 767 if len(error_msgs) >...

sinAshish