Skip to content

Commit 1e160ae

Browse files
author
JYLiu
committedMay 12, 2023
add model.eval() to test_aves.py
1 parent d1f172a commit 1e160ae

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed
 

‎test_aves.py

+2
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,8 @@ def forward(self, x, y=None):
3535
model_path='./aves-base-bio.pt',
3636
num_classes=10)
3737

38+
model.eval()
39+
3840
# Create a 1-second random sound
3941
waveform = torch.rand((16_000))
4042
x = waveform.unsqueeze(0)

0 commit comments

Comments
 (0)
Please sign in to comment.