Skip to content

Using next() instead of .next() in trainloaders #6

@cs-mshah

Description

@cs-mshah

The .next() should be changed to the following (stackoverflow):

model.train()
for batch_idx in range(args.train_iteration):
    try:
        inputs_x, targets_x, _ = next(labeled_train_iter)
    except:
        labeled_train_iter = iter(labeled_trainloader)
        inputs_x, targets_x, _ = next(labeled_train_iter)

    try:
        (inputs_u, inputs_u2), _, _ = next(unlabeled_train_iter)
    except:
        unlabeled_train_iter = iter(unlabeled_trainloader)
        (inputs_u, inputs_u2), _, _ = next(unlabeled_train_iter)

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