Update model.py

model.py

@@ -103,7 +103,7 @@ class LidirlCNN(PreTrainedModel):
     
         self.encoder = ConvolutionalBlock(config.embed_dim, config.channels, config.kernels, config.strides)
         self.embed_layer = nn.Embedding(config.vocab_size, config.embed_dim)
-        self.proj = ProjectionLayer(self.encoder.output_dim, label_size, montecarlo_layer)
+        self.proj = ProjectionLayer(self.encoder.output_dim, config.label_size, config.montecarlo_layer)
 
         self.label_size = config.label_size
         self.max_length = config.max_length