mirror of
https://github.com/MLSysBook/TinyTorch.git
synced 2026-06-03 12:00:55 -05:00
- Created perceptron_trained.py milestone with full training loop - Restored tinytorch/core/optimizers.py with Optimizer, SGD, Adam, AdamW classes - Fixed imports to use tinytorch.core.* instead of tensor_dev - Fixed tinytorch/core/losses.py with all loss functions - Fixed tinytorch/core/training.py imports ISSUE: Training loop runs but doesn't learn (gradients not flowing) - Loss stays constant at 0.7911 - Weights don't update - Likely autograd (Module 05) backward() not fully implemented - Need to fix Tensor.backward() and gradient computation
15 lines
337 B
Python
15 lines
337 B
Python
# AUTOGENERATED! DO NOT EDIT! File to edit: ../../modules/source/07_training/training_dev.ipynb.
|
|
|
|
# %% auto 0
|
|
__all__ = []
|
|
|
|
# %% ../../modules/source/07_training/training_dev.ipynb 1
|
|
import numpy as np
|
|
import pickle
|
|
import time
|
|
from typing import Dict, List, Optional, Tuple, Any, Callable
|
|
from pathlib import Path
|
|
import sys
|
|
import os
|
|
|