Zero'ing params docs

This commit is contained in:
Michael Cusack
2023-08-04 17:31:31 +07:00
parent 34f0402501
commit dfff7812db
-2
View File
@@ -1,4 +1,3 @@
#!/usr/bin/env python3
#!/usr/bin/env python
"""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():
p.detach().zero_()
"""
import glob
import os
import sys