gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-08-04 18:57:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: fix model input shape bug and costeer_model bug (#821)

Browse Source 
* fix model input shape bug and costeer_model bug

* fix a bug
This commit is contained in:
Yuante Li
2025-04-24 12:48:25 +08:00
committed by GitHub
parent c5beaeeaf8
commit ff99e4dc2c
3 changed files with 5 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/components/coder/model_coder/model_execute_template_v1.txt
+2 -2
View File
@@ -16,8 +16,8 @@ if MODEL_TYPE == "Tabular":
m = model_cls(num_features=input_shape[1])
data = torch.full(input_shape, INPUT_VALUE)
elif MODEL_TYPE == "TimeSeries":
input_shape = (BATCH_SIZE, NUM_FEATURES, NUM_TIMESTEPS)
m = model_cls(num_features=input_shape[1], num_timesteps=input_shape[2])
input_shape = (BATCH_SIZE, NUM_TIMESTEPS, NUM_FEATURES)
m = model_cls(num_features=input_shape[2], num_timesteps=input_shape[1])
data = torch.full(input_shape, INPUT_VALUE)
elif MODEL_TYPE == "Graph":
node_feature = torch.randn(BATCH_SIZE, NUM_FEATURES)