Skip to content
Snippets Groups Projects
Commit 8f621043 authored by Paul Best's avatar Paul Best
Browse files

fix Mel and humpback

parent 15bb7524
No related branches found
No related tags found
No related merge requests found
This diff is collapsed.
......@@ -86,7 +86,7 @@ frontend = {
STFT(nfft, int((sampleDur*sr - nfft)/128)),
Log1p(7, trainable=False),
nn.InstanceNorm2d(1),
u.Croper2D(n_mel, 128)
nn.AdaptiveMaxPool2d((128, 128))
),
'pcenMel': lambda sr, nfft, sampleDur, n_mel : nn.Sequential(
STFT(nfft, int((sampleDur*sr - nfft)/128)),
......
......@@ -15,7 +15,7 @@ parser.add_argument("-encoder", type=str, default='sparrow_encoder')
parser.add_argument("-frontend", type=str, default='logMel')
args = parser.parse_args()
modelname = f'{args.specie}_{args.bottleneck}_{args.frontend}{args.nMel}_{args.encoder}_decod2_BN_nomaxPool.stdc'
modelname = f'{args.specie}_{args.bottleneck}_{args.frontend}{args.nMel if "Mel" in args.frontend else ""}_{args.encoder}_decod2_BN_nomaxPool.stdc'
meta = models.meta[args.specie]
df = pd.read_csv(f'{args.specie}/{args.specie}.csv')
print(f'Tests for model {modelname}')
......
......@@ -16,7 +16,7 @@ parser.add_argument("specie", type=str)
parser.add_argument("-bottleneck", type=int, default=16)
parser.add_argument("-frontend", type=str, default='logMel')
parser.add_argument("-encoder", type=str, default='sparrow_encoder')
parser.add_argument("-nMel", type=int, default=64)
parser.add_argument("-nMel", type=int, default=128)
parser.add_argument("-lr", type=float, default=3e-3)
parser.add_argument("-lr_decay", type=float, default=1e-2)
parser.add_argument("-batch_size", type=int, default=128)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment