Sadjad Alikhani
commited on
Commit
•
84e536e
1
Parent(s):
48ca955
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]
|