minor fixes in model to avoid downloading pretrained weights for resnet backbone model

iam_line_recognition/model_main.py

@@ -87,6 +87,7 @@ class CRNN(nn.Module):
         image_height,
         num_feats_mapped_seq_hidden=128,
         num_feats_seq_hidden=256,
+        pretrained=False,
     ):
         """
         ---------
@@ -102,7 +103,7 @@ class CRNN(nn.Module):
             number of features to be used in the LSTM for sequence modeling (default: 256)
         """
         super().__init__()
-        self.visual_feature_extractor = ResNetFeatureExtractor()
+        self.visual_feature_extractor = ResNetFeatureExtractor(pretrained=pretrained)
         self.rnn_seq2seq_module = HW_RNN_Seq2Seq(
             num_classes,
             image_height,
@@ -134,6 +135,7 @@ class STN_CRNN(nn.Module):
         image_width,
         num_feats_mapped_seq_hidden=128,
         num_feats_seq_hidden=256,
+        pretrained=False,
     ):
         """
         ---------
@@ -157,7 +159,7 @@ class STN_CRNN(nn.Module):
             (image_height, image_width),
             I_channel_num=3,
         )
-        self.visual_feature_extractor = ResNetFeatureExtractor()
+        self.visual_feature_extractor = ResNetFeatureExtractor(pretrained=pretrained)
         self.rnn_seq2seq_module = HW_RNN_Seq2Seq(
             num_classes,
             image_height,