Skip to content
Snippets Groups Projects
Commit bf4a24ef authored by Cassandra Grzonkowski's avatar Cassandra Grzonkowski
Browse files

load indexed vocabulary in init danceformer

parent ea0ef823
No related branches found
No related tags found
1 merge request!1Instant spectograms
......@@ -397,7 +397,7 @@ if __name__ == '__main__':
# initializes MusicNet, MlP/DifficultyNet, transformer
danceformer = Danceformer().to(device)
danceformer = Danceformer(indexed_vocabulary).to(device)
print(danceformer)
# number parameter
print("Number parameters:")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment