Sadjad Alikhani commited on
Commit
84e536e
1 Parent(s): 48ca955

Update input_preprocess.py

Browse files
Files changed (1) hide show
  1. input_preprocess.py +2 -2
input_preprocess.py CHANGED
@@ -41,8 +41,8 @@ def tokenizer(selected_scenario_names=None, manual_data=None, gen_raw=True):
41
  """
42
 
43
  if manual_data is not None:
44
- #patches = patch_maker(np.expand_dims(np.array(manual_data), axis=1))
45
- patches = patch_maker(torch.tensor(manual_data, dtype=torch.complex64).unsqueeze(1))
46
  else:
47
  # Patch generation or loading
48
  deepmimo_data = [DeepMIMO_data_gen(scenario_name) for scenario_name in selected_scenario_names]
 
41
  """
42
 
43
  if manual_data is not None:
44
+ patches = patch_maker(np.expand_dims(np.array(manual_data), axis=1))
45
+ #patches = patch_maker(torch.tensor(manual_data, dtype=torch.complex64).unsqueeze(1))
46
  else:
47
  # Patch generation or loading
48
  deepmimo_data = [DeepMIMO_data_gen(scenario_name) for scenario_name in selected_scenario_names]