1eeb316d66b5ab7c377f1926fc24302dd1e9d418,VAEs/vq-vae/models/vq_vae.py,Model,encoder,#Model#Any#Any#,186

Before Change

				pad=(1,1), name = "conv_2", rng=self.rng)
			out = F.relu(out)

			out = PF.convolution(out, self.num_hidden, (3,3), stride=(1,1),
				pad=(1,1), name = "conv_3", rng=self.rng)

			out = self.encoder_res_stack(out)	
		if np.any(np.isnan(out.d)):
			import pdb; pdb.set_trace()		

After Change

		with nn.parameter_scope("encoder"):
			out = PF.convolution(x, self.num_hidden, (4,4), stride=(2,2),
				pad=(1,1), name = "conv_1", rng=self.rng)
			out = PF.batch_normalization(out)
			out = F.relu(out)
			out = PF.convolution(out, self.num_hidden, (4,4), stride=(2,2),
				pad=(1,1), name = "conv_2", rng=self.rng)

Instances