HF RL Explorer

Task: Calculate the Number of Parameters in VGG-16 Neural Network

Task: Calculate the Number of Parameters in VGG-16 Neural Network: a task in Terminal-Lego-15k (Harbor dataset). Write a Python script that correctly calculates the total number of trainable parameters in the VGG-16 convolutional neural network architecture, broken down by each layer.

The task

Write a Python script that correctly calculates the total number of trainable parameters in the VGG-16 convolutional neural network architecture, broken down by each layer.

Part of PrimeIntellect/Terminal-Lego-15k.