MCPcopy Create free account
hub / github.com/NVIDIA/TensorRT / get_weights

Method get_weights

samples/python/network_api_pytorch_mnist/model.py:129–130  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

127 test(e + 1)
128
129 def get_weights(self):
130 return self.network.state_dict()
131
132 def get_random_testcase(self):
133 data, target = next(iter(self.test_loader))

Callers 5

mainFunction · 0.95
_execute_quantize_modelFunction · 0.80

Calls

no outgoing calls