0fd13ccd7e61c9186127993080240c07b7ad905e,docs/tutorials/action_recognition/dive_deep_ucf101.py,,,#,63

Before Change

        // AutoGrad
        with ag.record():
            output = [net(X) for X in data]
            loss = [loss_fn(yhat, y) for yhat, y in zip(output, label)]

        // Backpropagation

After Change

            output = []
            for _, X in enumerate(data):
                X = X.reshape((-1,) + X.shape[2:])
                pred = net(X)
                output.append(pred)
            loss = [loss_fn(yhat, y) for yhat, y in zip(output, label)]

        // Backpropagation

Instances