Anpassungen preTrained

preTrainedTest.py

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