Spaces:
Running
Running
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
|