test : main_modified3.py
please import  torch.nn.utils.prune
torch.nn.utils.prune:  https://pytorch.org/tutorials/intermediate/pruning_tutorial.html

# Environment
```shell
python==3.9
pytorch==1.9.0
```
if you want choose different non-linear reg:
please give a new list cover "weigth_decay = []"