Describe the bug
save/load does not work anymore when using TCNModel with weight_norm=True, Pytorch ask for model.state_dict()

probably related to this update
Replaced the deprecated torch.nn.utils.weight_norm function with torch.nn.utils.parametrizations.weight_norm. #2593 by Saeed Foroutan.