Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
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
main
Function · 0.95
_execute_quantize_model
Function · 0.80
_print_model_weights_shapes
Function · 0.80
test_clone_numerics_quantize_whole_model
Function · 0.80
print_model_weights_shapes
Function · 0.80
Calls
no outgoing calls
Tested by
2
_print_model_weights_shapes
Function · 0.64
test_clone_numerics_quantize_whole_model
Function · 0.64