10000 DataLoader in function of load_dataset in utils.py · Issue #22 · chaoshangcs/GTS · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

DataLoader in function of load_dataset in utils.py #22

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you 8000 agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Open
fuyuyuputao opened this issue Jul 20, 2022 · 1 comment
Open

DataLoader in function of load_dataset in utils.py #22

fuyuyuputao opened this issue Jul 20, 2022 · 1 comment

Comments

@fuyuyuputao
Copy link 
data['train_loader'] = DataLoader(data['x_train'], data['y_train'], batch_size, shuffle=True)
data['val_loader'] = DataLoader(data['x_val'], data['y_val'], test_batch_size, shuffle=False)
data['test_loader'] = DataLoader(data['x_test'], data['y_test'], test_batch_size, shuffle=False)

Hi! Maybe there is a very small error. I think the second row about val_loader is "val_batch_size" not "test_batch_size" even if these two size are equal 64.
@chaoshangcs
Copy link
Owner

Hi, thanks for your message. Here we only used one variable "test_batch_size" for both the valid dataloader and test dataloader. You provided a great suggestion. We could use two variables here for more clarification. Thanks!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@chaoshangcs @fuyuyuputao
0