From 808a1a4a310e6d885785139b3d2aae457eb6285f Mon Sep 17 00:00:00 2001 From: James Betker Date: Mon, 4 Jul 2022 08:50:09 -0600 Subject: [PATCH] eval cheater_ar --- codes/utils/music_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/utils/music_utils.py b/codes/utils/music_utils.py index 9161c386..788c1bb5 100644 --- a/codes/utils/music_utils.py +++ b/codes/utils/music_utils.py @@ -56,5 +56,5 @@ def get_ar_prior(): from models.audio.music.cheater_gen_ar import ConditioningAR cheater_ar = ConditioningAR(1024, layers=24, dropout=0, cond_free_percent=0) cheater_ar.load_state_dict(torch.load('../experiments/music_cheater_ar.pth', map_location=torch.device('cpu'))) - #cheater_ar = cheater_ar.eval() + cheater_ar = cheater_ar.eval() return cheater_ar \ No newline at end of file