From 6d50a394030f6620109e00faf12014af9a5bb4fe Mon Sep 17 00:00:00 2001
From: Paul Best <paul.best@lis-lab.fr>
Date: Wed, 28 Sep 2022 10:26:10 +0200
Subject: [PATCH] int in resample
---
run_CNN.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/run_CNN.py b/run_CNN.py
index 3f5c2b5..4b25fe5 100644
--- a/run_CNN.py
+++ b/run_CNN.py
@@ -50,7 +50,7 @@ class Dataset(torch.utils.data.Dataset):
return None
sig = sig[:, args.channel]
if fs != models.get[args.specie]['fs']:
- sig = signal.resample(sig, args.lensample * models.get[args.specie]['fs'])
+ sig = signal.resample(sig, int(args.lensample * models.get[args.specie]['fs']))
sig = norm(sig)
return torch.tensor(sig).float(), sample
--
GitLab