deviandice/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

eval cheater_ar

Browse Source
This commit is contained in:
James Betker 2022-07-04 08:50:09 -06:00
parent bac9a8b728
commit 808a1a4a31
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/utils/music_utils.py
View File

@ -56,5 +56,5 @@ def get_ar_prior():
from models.audio.music.cheater_gen_ar import ConditioningAR from models.audio.music.cheater_gen_ar import ConditioningAR
cheater_ar = ConditioningAR(1024, layers=24, dropout=0, cond_free_percent=0) 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.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 return cheater_ar