debug

.idea/workspace.xml

@@ -6,7 +6,7 @@
   <component name="ChangeListManager">
     <list default="true" id="23565123-73ab-4f40-a9ef-1086e0c9e1ec" name="Changes" comment="">
       <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/script.py" beforeDir="false" afterPath="$PROJECT_DIR$/script.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/preprocess.py" beforeDir="false" afterPath="$PROJECT_DIR$/preprocess.py" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -99,7 +99,7 @@
       <updated>1742573353560</updated>
       <workItem from="1742573355153" duration="587000" />
       <workItem from="1742806974298" duration="2741000" />
-      <workItem from="1742810431420" duration="13956000" />
+      <workItem from="1742810431420" duration="14639000" />
     </task>
     <servers />
   </component>

preprocess.py

@@ -33,6 +33,7 @@ def preprocess(audio_file):
         frames += [y[int(start_sample):int(start_sample) + win_len]]
     y_win = np.stack(frames, axis=0)
     tensor = torch.from_numpy(y_win)
+    tensor = tensor.float()
 
     print(f'preprocessed track - shape {tensor.shape}')
     return tensor