From 93c4b08c433bf7a0016581663a146cb541b4eef9 Mon Sep 17 00:00:00 2001
From: Paul Best <paul.best@lis-lab.fr>
Date: Tue, 20 Jun 2023 11:31:11 +0200
Subject: [PATCH] Update file train_AE.py
---
new_specie/train_AE.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/new_specie/train_AE.py b/new_specie/train_AE.py
index 15f63eb..745c76f 100755
--- a/new_specie/train_AE.py
+++ b/new_specie/train_AE.py
@@ -38,7 +38,7 @@ loss_fun = torch.nn.MSELoss()
df = pd.read_csv(args.detections)
loader = torch.utils.data.DataLoader(u.Dataset(df, args.audio_folder, args.SR, args.sampleDur), batch_size=batch_size, shuffle=True, num_workers=8, collate_fn=u.collate_fn)
-modelname = f'{args.detections[:-4]}_AE_{args.bottleneck}_mel{args.nMel}.stdc'
+modelname = f'{args.detections[:-4]}_AE_{args.bottleneck}_mel{args.nMel}.weights'
step, writer = 0, SummaryWriter('runs/'+modelname)
print(f'Go for model {modelname} with {len(df)} vocalizations')
for epoch in range(100_000//len(loader)):
--
GitLab