b5bf0e5eb1ddef22615e62b90a12898bdd639be9,test/test_nn.py,TestNet,test_trainable,#TestNet#Any#,15

Before Change

        print("Running check_trainable test:")
        flag = True
        before_params = net.gather_trainable_params()
        dummy_input = Variable(torch.ones((2, net.in_dim)))
        dummy_output = Variable(torch.zeros((2, net.out_dim)))
        loss = net.training_step(dummy_input, dummy_output)
        after_params = net.gather_trainable_params()
        i = 0

After Change

        net = test_nets[0]
        flag = True
        before_params = net.gather_trainable_params()
        if type(net.in_dim) is int:
            dummy_input = Variable(torch.ones(2, net.in_dim))
        else:
            dummy_input = Variable(torch.ones(2, *net.in_dim))
        dummy_output = Variable(torch.zeros((2, net.out_dim)))
        loss = net.training_step(dummy_input, dummy_output)
        after_params = net.gather_trainable_params()
        i = 0

Instances