diff --git a/new_specie/train_AE.py b/new_specie/train_AE.py
index 15f63ebe99eb00e0d2a958b21476a1c60e6a673d..745c76f2e8af97dbb77384d49dccf4a6d5dbd5fd 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)):