8e9a026bc34d8409faa572a7144c2d96a7c039ed,tests/models/test_hparams.py,SimpleNoArgsModel,test_step,#SimpleNoArgsModel#Any#Any#,545

Before Change

    def test_step(self, batch, batch_nb):
        x, y = batch
        loss = F.cross_entropy(self(x), y)
        return {"loss": loss, "log": {"train_loss": loss}}

    def configure_optimizers(self):
        return torch.optim.Adam(self.parameters(), lr=0.02)

After Change

// -------------------------
// STANDARD TESTS
// -------------------------
def _run_standard_hparams_test(tmpdir, model, cls, try_overwrite=False):
    
    Tests for the existence of an arg "test_arg=14"

Instances