BMP commited on
Commit
18c9410
·
verified ·
1 Parent(s): e866cd9

Upload usage_example.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. usage_example.py +2 -1
usage_example.py CHANGED
@@ -13,7 +13,8 @@ def load_model(model_path="."):
13
  # Initialize model
14
  model = CAMPPModel(
15
  input_dim=config["input_dim"],
16
- embedding_dim=config["embedding_dim"]
 
17
  )
18
 
19
  # Load weights
 
13
  # Initialize model
14
  model = CAMPPModel(
15
  input_dim=config["input_dim"],
16
+ embedding_dim=config["embedding_dim"],
17
+ input_channels=config.get("input_channels", 64)
18
  )
19
 
20
  # Load weights