CUDA runs out of memory while trying to build the wide resnet models from the example script e2wrn.py

[apogee4286]

When I tried to instantiate a model using the line: model = wrn28_10_d8d4d1(num_classes=10), I could see in the terminal that building the layers became increasingly slow as their size increased to the point where my cuda just ran out of memory. Am I doing something wrong? My computer has an rtx3070 ti with a ryzen 7 cpu and 32 gb ram.