From a8ac362121036656954f323ffcf60112ade3a902 Mon Sep 17 00:00:00 2001 From: Eddie Date: Sun, 6 Feb 2022 18:24:59 +0200 Subject: [PATCH] remove typo in model definition --- model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model.py b/model.py index 3760e88..47cf387 100644 --- a/model.py +++ b/model.py @@ -11,7 +11,7 @@ class MS_TCN2(nn.Module): -MS_TCB def __init__(self, num_layers_PG, num_layers_R, num_R, num_f_maps, dim, num_classes): + def __init__(self, num_layers_PG, num_layers_R, num_R, num_f_maps, dim, num_classes): super(MS_TCN2, self).__init__() self.PG = Prediction_Generation(num_layers_PG, num_f_maps, dim, num_classes) self.Rs = nn.ModuleList([copy.deepcopy(Refinement(num_layers_R, num_f_maps, num_classes, num_classes)) for s in range(num_R)])