Skip to content

There is a mistake in shuffle_minibatch function #3

@tmdrn9

Description

@tmdrn9

Hello.
Thank you for your share.

First of all, My English is not so fluent.
Please understand it.

There is a mistake in shuffle_minibatch function in main_cifar10.py, main_cifar100.py

def shuffle_minibatch(inputs, targets, n_class=88, mixup=True):
    """Shuffle a minibatch and do linear interpolation between images and labels.
    Args:
        inputs: a numpy array of images with size batch_size x H x W x 3.
        targets: a numpy array of labels with size batch_size x 1.
        mixup: a boolen as whether to do mixup or not. If mixup is True, we
            sample the weight from beta distribution using parameter alpha=1,
            beta=1. If mixup is False, we set the weight to be 1 and 0
            respectively for the randomly shuffled mini-batches.
    """

According to your comment, inputs parameter shape is batch_size x H x W x 3.
but your input shape is batch_sizex 3 x H x W, isn't it?

I think you change comment

Thank you for your work again.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions