Spaces:
Running
Running
add "num_patch_input=7, padding_mode='constant'" in model.update_config()
Browse files
app.py
CHANGED
|
@@ -211,7 +211,9 @@ def predict_at_index(df, index, context_len, pred_len):
|
|
| 211 |
periodicity = periodicity_list[0] if periodicity_list else 1
|
| 212 |
|
| 213 |
color_list = [i % 3 for i in range(nvars)]
|
| 214 |
-
model.update_config(context_len=context_len, pred_len=pred_len, periodicity=periodicity)
|
|
|
|
|
|
|
| 215 |
|
| 216 |
with torch.no_grad():
|
| 217 |
y_pred, input_image, reconstructed_image, _, _ = model.forward(
|
|
@@ -225,7 +227,7 @@ def predict_at_index(df, index, context_len, pred_len):
|
|
| 225 |
print(f"{reconstructed_image.shape = }")
|
| 226 |
print(f"{len(y_pred_quantile_list) = }")
|
| 227 |
|
| 228 |
-
print(f"{input_image = }")
|
| 229 |
print(f"{input_image[0,0,0, :, 0] = }")
|
| 230 |
print(f"{input_image[0,0,0, 50:70, 0] = }")
|
| 231 |
print(f"{input_image[0,0,0, 100:120, 0] = }")
|
|
|
|
| 211 |
periodicity = periodicity_list[0] if periodicity_list else 1
|
| 212 |
|
| 213 |
color_list = [i % 3 for i in range(nvars)]
|
| 214 |
+
# model.update_config(context_len=context_len, pred_len=pred_len, periodicity=periodicity)
|
| 215 |
+
model.update_config(context_len=context_len, pred_len=pred_len, periodicity=periodicity,
|
| 216 |
+
num_patch_input=7, padding_mode='constant')
|
| 217 |
|
| 218 |
with torch.no_grad():
|
| 219 |
y_pred, input_image, reconstructed_image, _, _ = model.forward(
|
|
|
|
| 227 |
print(f"{reconstructed_image.shape = }")
|
| 228 |
print(f"{len(y_pred_quantile_list) = }")
|
| 229 |
|
| 230 |
+
# print(f"{input_image = }")
|
| 231 |
print(f"{input_image[0,0,0, :, 0] = }")
|
| 232 |
print(f"{input_image[0,0,0, 50:70, 0] = }")
|
| 233 |
print(f"{input_image[0,0,0, 100:120, 0] = }")
|