N
NeuronLabs

Practice Sandbox: pytorch

Free Practice Mode

🔥 Model Graph

ResNet-50
Conv2d(3, 64, kernel_size=(7, 7), stride=(2, 2))
Parameters: 9,408
BatchNorm2d(64, eps=1e-05)
Parameters: 128
ReLU(inplace=True)
Parameters: 0
MaxPool2d(kernel_size=3, stride=2, padding=1)
...

Live Training

Epoch14/50
[INFO] Loading dataset (CIFAR-10) onto GPU:0... OK
[INFO] Starting mixed precision training loop.
Epoch 12/50: loss=0.4512, acc=86.2%
Epoch 13/50: loss=0.4201, acc=87.5%
Epoch 14/50: loss=0.3890, acc=88.1% [BATCH 450/1500]
30%

Hardware Acceleration

NVIDIA A100-SXM4 (GPU:0)98%
VRAM Usage32GB / 40GB
Temp: 78°CPwr: 380W

Hyperparameters