github-starred/TinyTorch
2
0
Fork 0
mirror of https://github.com/MLSysBook/TinyTorch.git synced 2026-03-11 23:53:33 -05:00
Code Issues 7 Packages Projects Releases Wiki Activity

Fix imports: Replace dev-style imports with proper package imports in modules 06-07

Browse Source
This commit is contained in:
Vijay Janapa Reddi
2025-09-30 13:40:38 -04:00
parent c0a1dd257a
commit ae3f2246d9
2 changed files with 5 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
modules/source/06_optimizers/optimizers_dev.py
View File

@@ -63,10 +63,7 @@ import numpy as np
from typing import List, Union, Optional, Dict, Any
# Import Tensor from Module 01 (now with gradient support from Module 05)
import sys
import os
sys.path.append(os.path.join(os.path.dirname(__file__), '..', '01_tensor'))
from tensor_dev import Tensor
from tinytorch.core.tensor import Tensor
# %% [markdown]
"""

15
modules/source/07_training/training_dev.py
View File

@@ -69,17 +69,10 @@ import sys
import os
# Import dependencies from other modules
sys.path.append(os.path.join(os.path.dirname(__file__), '..', '01_tensor'))
from tensor_dev import Tensor
sys.path.append(os.path.join(os.path.dirname(__file__), '..', '03_layers'))
from layers_dev import Linear
sys.path.append(os.path.join(os.path.dirname(__file__), '..', '04_losses'))
from losses_dev import MSELoss, CrossEntropyLoss
sys.path.append(os.path.join(os.path.dirname(__file__), '..', '06_optimizers'))
from optimizers_dev import SGD, AdamW
from tinytorch.core.tensor import Tensor
from tinytorch.core.layers import Linear
from tinytorch.core.losses import MSELoss, CrossEntropyLoss
from tinytorch.core.optimizers import SGD, AdamW
# %% [markdown]
"""