Skip to content
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.

By clicking “Sign up for GitHub”, you 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

Function arguments of get_loader_btcv in test.py #15

Open
ToruHironaka opened this issue May 12, 2023 · 1 comment
Open

Function arguments of get_loader_btcv in test.py #15

ToruHironaka opened this issue May 12, 2023 · 1 comment

Comments

@ToruHironaka
Copy link

BTCV test.py script's code below at line 151. The function get_loader_btcv's arguments set batch_size, but this function parameter does not have batch_size. Here is the function parameter get_loader_btcv(data_dir, cache=True). I assume data_dir=./RawData/Training because get_loader_btcv read image and label files and RawData/Testing has only img directory. Am I setting the function arguments correct? or Am I using a wrong dataset? Please let me know.

train_ds, val_ds, test_ds = get_loader_btcv(batch_size=batch_size, fold=0, cache=False)

@920232796
Copy link
Contributor

I have fixed the error, thank you.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants