Skip to content

Commit 9f7e5b0

Browse files
committed
[ADD] minor change to specify that partition is a dataset partition, perhaps that's not obvious
1 parent 40793a2 commit 9f7e5b0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

lstm_genre_classifier_pytorch_lightning.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -105,8 +105,8 @@ def configure_optimizers(self):
105105

106106
class GTZANDataset(data.Dataset):
107107

108-
def __init__(self, partition) -> object:
109-
self.partition = partition
108+
def __init__(self, dataset_partition) -> object:
109+
self.partition = dataset_partition
110110
self.genre_features = GenreFeatureData()
111111

112112
# if all of the preprocessed files do not exist, regenerate them all for self-consistency

0 commit comments

Comments
 (0)