Anpassungen modelpath

preTrainedTest.py

@@ -12,11 +12,12 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 print(f"Using device: {device}")
 
 # Modell laden
-model_name_or_path = "TimM77/SegformerPlusPlus"
+config_path = "TimM77/SegformerPlusPlus"
 print("Starte config_load")
-config = MySegformerConfig.from_pretrained(model_name_or_path)
+config = MySegformerConfig.from_pretrained(config_path)
 print("Starte Model_load")
-model = MySegformerForSemanticSegmentation.from_pretrained(model_name_or_path, config=config)
+model_path = "TimM77/SegformerPlusPlus/pytorch_model.bin"
+model = MySegformerForSemanticSegmentation.from_pretrained(model_path, config=config)
 model.to(device).eval()
 
 # Bild laden