Hi Jin, thanks for good work,
I am wondering how to prepare training dataset file and train the model
Could you share your ""train_val_test_paths_labels.pkl file to prepare for custom data set.
def main():
train_dataset, train_num_each, val_dataset, val_num_each, _, _ = get_data('train_val_test_paths_labels.pkl')
train_model(train_dataset, train_num_each, val_dataset, val_num_each)
Thanks for your support