HachiML commited on
Commit
cffd8cf
1 Parent(s): f2ce0c1

Update modeling_mists.py

Browse files
Files changed (1) hide show
  1. modeling_mists.py +5 -5
modeling_mists.py CHANGED
@@ -241,11 +241,11 @@ class MistsForConditionalGeneration(MistsPreTrainedModel):
241
 
242
  # モデルInput確認用(学習時)
243
  # print("model inputs:")
244
- print("input_ids: ", input_ids.shape if input_ids is not None else None)
245
- print("attention_mask: ", attention_mask.shape if attention_mask is not None else None)
246
- print("time_series_values: ", time_series_values.shape if time_series_values is not None else None)
247
- print("time_series_input_mask: ", time_series_input_mask.shape if time_series_input_mask is not None else None)
248
- print("==========")
249
 
250
  if inputs_embeds is None:
251
  # 1. Extra the input embeddings
 
241
 
242
  # モデルInput確認用(学習時)
243
  # print("model inputs:")
244
+ # print("input_ids: ", input_ids.shape if input_ids is not None else None)
245
+ # print("attention_mask: ", attention_mask.shape if attention_mask is not None else None)
246
+ # print("time_series_values: ", time_series_values.shape if time_series_values is not None else None)
247
+ # print("time_series_input_mask: ", time_series_input_mask.shape if time_series_input_mask is not None else None)
248
+ # print("==========")
249
 
250
  if inputs_embeds is None:
251
  # 1. Extra the input embeddings