(network, path='models')
| 32 | |
| 33 | |
| 34 | def restore_params(network, path='models'): |
| 35 | logging.info("Restore pre-trained parameters") |
| 36 | maybe_download_and_extract( |
| 37 | 'squeezenet.npz', path, 'https://github.com/tensorlayer/pretrained-models/raw/master/models/', |
| 38 | expected_bytes=7405613 |
| 39 | ) # ls -al |
| 40 | params = load_npz(name=os.path.join(path, 'squeezenet.npz')) |
| 41 | assign_weights(params[:len(network.all_weights)], network) |
| 42 | del params |
| 43 | |
| 44 | |
| 45 | def SqueezeNetV1(pretrained=False, end_with='out', name=None): |
no test coverage detected
searching dependent graphs…