Lefei commited on
Commit
e8ad305
·
verified ·
1 Parent(s): af67167

add "num_patch_input=7, padding_mode='constant'" in model.update_config()

Browse files
Files changed (1) hide show
  1. app.py +4 -2
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] = }")