Sadjad Alikhani
commited on
Update input_preprocess.py
Browse files- 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 |
-
|
| 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]
|