Zero'ing params docs
This commit is contained in:
@@ -1,4 +1,3 @@
|
|||||||
#!/usr/bin/env python3
|
|
||||||
#!/usr/bin/env python
|
#!/usr/bin/env python
|
||||||
"""Saves the model as a TorchScript.
|
"""Saves the model as a TorchScript.
|
||||||
|
|
||||||
@@ -14,7 +13,6 @@ parameters before saving it and it will gzip down to 780K:
|
|||||||
for p in model.parameters():
|
for p in model.parameters():
|
||||||
p.detach().zero_()
|
p.detach().zero_()
|
||||||
"""
|
"""
|
||||||
|
|
||||||
import glob
|
import glob
|
||||||
import os
|
import os
|
||||||
import sys
|
import sys
|
||||||
|
|||||||
Reference in New Issue
Block a user