error in 10_Utilities

[kumar-mahendra]

class Dataset 's method collate_fn needs a little change as otherwise following error in thrown when creating dataloader

ValueError: setting an array element with a sequence

Given Code

"""Processing on a batch."""
    # Get inputs
    batch = np.array(batch, dtype=object)
    X = batch[:, 0]    # This line execution throws above error 
    y = np.stack(batch[:, 1], axis=0)

Suggested solution

"""Processing on a batch."""
    # Get inputs
    batch = np.array(batch, dtype=object)
    X = np.stack(batch[:, 0] ,axis=0) 
    y = np.stack(batch[:, 1], axis=0)

[GokuMohandas]

Nice catch Kumar! Since our inputs are of the same length I have to stack them here. In subsequent lessons with variable sequence lengths within a batch, we have to use batch[:, i]. I'll push this fix later today.