MCPcopy
hub / github.com/tensorlayer/TensorLayer / restore_params

Function restore_params

tensorlayer/models/squeezenetv1.py:34–42  ·  view source on GitHub ↗
(network, path='models')

Source from the content-addressed store, hash-verified

32
33
34def 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
45def SqueezeNetV1(pretrained=False, end_with='out', name=None):

Callers 1

SqueezeNetV1Function · 0.70

Calls 3

load_npzFunction · 0.90
assign_weightsFunction · 0.90

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…