CIS6270 / cis6270 /__init__.py
pranamanam's picture
Upload 87 files
0ba9d09 verified
Raw History Blame Contribute Delete
725 Bytes
"""Shared helpers for the CIS 6270 course code.
The data, the small networks, the training loop and the sample metrics live here so that each
lecture file holds only the mathematics of its own framework. Read this package once; after that
every lecture file stands on its own.
"""
import warnings
# torch 2.14 warns about `torch.jit.script` from inside its own forward-mode autograd code, the
# first time `torch.func.jvp` runs. Chapter 6 runs that function, so we filter the message here.
warnings.filterwarnings(
"ignore", message=r"`torch\.jit\.script` is deprecated", category=FutureWarning,
)
from cis6270 import data, metrics, nets, runner
__all__ = ["data", "metrics", "nets", "runner"]
__version__ = "1.0.0"