HF RL Explorer

Task: Visualize a PyTorch Neural Network using torchviz

Task: Visualize a PyTorch Neural Network using torchviz: a task in Terminal-Lego-15k (Harbor dataset). Write a Python script that correctly visualizes a pretrained ResNet-50 model from PyTorch's torchvision library using the torchviz package. The script must generate a computational graph…

The task

Write a Python script that correctly visualizes a pretrained ResNet-50 model from PyTorch's torchvision library using the `torchviz` package. The script must generate a computational graph visualization and save it as a PDF file.

Part of PrimeIntellect/Terminal-Lego-15k.