Princess3 commited on
Commit
02e561f
1 Parent(s): 9a38883

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -284,7 +284,7 @@ def main():
284
  The model architecture is determined by XML files in the 'Xml_Data' folder,
285
  where each subfolder represents a model section containing layer configurations.
286
  """
287
- folder_path = 'Xml_Data'
288
  model = create_model_from_folder(folder_path)
289
 
290
  print(f"Created dynamic PyTorch model with sections: {list(model.sections.keys())}")
 
284
  The model architecture is determined by XML files in the 'Xml_Data' folder,
285
  where each subfolder represents a model section containing layer configurations.
286
  """
287
+ folder_path = 'data'
288
  model = create_model_from_folder(folder_path)
289
 
290
  print(f"Created dynamic PyTorch model with sections: {list(model.sections.keys())}")