mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-06-13 08:28:55 +08:00
56d44637f3
* fixed pylint, formatted python files iwth cblack on localhost
* Revert "fixed pylint, formatted python files iwth cblack on localhost"
This reverts commit 07e2b88466.
* dedented 4-spaces added linter
Co-authored-by: Iain Wong <iainwong@outlook.com>
28 lines
784 B
Python
28 lines
784 B
Python
from tinygrad.tensor import Tensor
|
|
|
|
def fetch(url):
|
|
import requests, os, hashlib, tempfile
|
|
fp = os.path.join(tempfile.gettempdir(), hashlib.md5(url.encode('utf-8')).hexdigest())
|
|
if os.path.isfile(fp) and os.stat(fp).st_size > 0:
|
|
with open(fp, "rb") as f:
|
|
dat = f.read()
|
|
else:
|
|
print("fetching %s" % url)
|
|
dat = requests.get(url).content
|
|
with open(fp+".tmp", "wb") as f:
|
|
f.write(dat)
|
|
os.rename(fp+".tmp", fp)
|
|
return dat
|
|
|
|
def get_parameters(obj):
|
|
parameters = []
|
|
if isinstance(obj, Tensor):
|
|
parameters.append(obj)
|
|
elif isinstance(obj, list):
|
|
for x in obj:
|
|
parameters.extend(get_parameters(x))
|
|
elif hasattr(obj, '__dict__'):
|
|
for k,v in obj.__dict__.items():
|
|
parameters.extend(get_parameters(v))
|
|
return parameters
|