Update config.py

config.py

@@ -22,11 +22,11 @@ epochs = 350
 gpu = 1
 device = torch.device(f"cuda:{gpu}" if torch.cuda.is_available() else "cpu")
 
-model_name = 'efficientnet_b2'
-image_embedding = 1408
+# model_name = 'efficientnet_b2'
+# image_embedding = 1408
 
-# model_name = 'resnet50'
-# image_embedding = 1000
+model_name = 'resnet50'
+image_embedding = 1000
 
 text_encoder_model = "neuralspace-reverie/indic-transformers-bn-bert" 
 text_tokenizer = "neuralspace-reverie/indic-transformers-bn-bert"