PyTorch Training Cost Estimator
Estimates per-epoch training cost for PyTorch model.
NN.Module (must be named Model)
Input Shape (must set variable tuple `input_shape`)
Loss Function
Optimizer
Training Examples Per Epoch
Target Hardware
T4
$0.59 / hr
A10
$1.10 / hr
A100 (40GB)
$2.10 / hr
H100
$3.95 / hr
H200
$4.54 / hr
Compiler Settings
Use torch.compile (recommended)
Don't use torch.compile
Estimate Training Cost
Examples
MLP
Alex Net