Spaces:
Running
Running
MilesCranmer
commited on
Commit
•
aefe008
1
Parent(s):
87f1eda
docs: clearer about torch port
Browse files- pysr/export_torch.py +2 -2
pysr/export_torch.py
CHANGED
@@ -82,7 +82,7 @@ def _initialize_torch():
|
|
82 |
}
|
83 |
|
84 |
class _Node(torch.nn.Module):
|
85 |
-
"""
|
86 |
|
87 |
def __init__(self, *, expr, _memodict, _func_lookup, **kwargs):
|
88 |
super().__init__(**kwargs)
|
@@ -154,7 +154,7 @@ def _initialize_torch():
|
|
154 |
return self._torch_func(*args)
|
155 |
|
156 |
class _SingleSymPyModule(torch.nn.Module):
|
157 |
-
"""
|
158 |
|
159 |
def __init__(
|
160 |
self, expression, symbols_in, selection=None, extra_funcs=None, **kwargs
|
|
|
82 |
}
|
83 |
|
84 |
class _Node(torch.nn.Module):
|
85 |
+
"""Forked from https://github.com/patrick-kidger/sympytorch"""
|
86 |
|
87 |
def __init__(self, *, expr, _memodict, _func_lookup, **kwargs):
|
88 |
super().__init__(**kwargs)
|
|
|
154 |
return self._torch_func(*args)
|
155 |
|
156 |
class _SingleSymPyModule(torch.nn.Module):
|
157 |
+
"""Forked from https://github.com/patrick-kidger/sympytorch"""
|
158 |
|
159 |
def __init__(
|
160 |
self, expression, symbols_in, selection=None, extra_funcs=None, **kwargs
|