Here is my code
dataset = pd.read_csv(\'augmented_data.csv\') dataset = dataset.sample(frac=1) class ConvNet(nn.Module): def __init__(self): super(ConvNet, s