Skip to content

Why in evaluation there are 1856 predictions #30

Discussion options

You must be logged in to vote

The dataloader with drop_last =True will drop the last items which can not make another batch.

In the baseline, the batchsize is set at 32, and 1873/32 = 58.53125, 1856/32 = 58. Thus the last 17 items will be drop.

        if flag == 'test':
            shuffle_flag = False
            drop_last = True
        else:
            shuffle_flag = True
            drop_last = True

        ......
        data_loader = DataLoader(
            data_set,
            batch_size=self.args["batch_size"],
            shuffle=shuffle_flag,
            num_workers=self.args["num_workers"],
            drop_last=drop_last
        )

Replies: 1 comment 8 replies

Comment options

You must be logged in to vote
8 replies
@zhoujingbo
Comment options

@phoenixes94
Comment options

@phoenixes94
Comment options

@ramber1836
Comment options

@phoenixes94
Comment options

Answer selected by zhoujingbo
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
6 participants