# Codeblock 1 import torch import torch.nn as nn GROWTH = 12 COMPRESSION = 0.5 REPEATS = [6, 12, 24, 16]