Commit 607b7ab

committed

specify contiguous before flattening reconstructions

1 parent f3344b7 commit 607b7abCopy full SHA for 607b7ab

File tree

-4

lines changed

-4

lines changed

Lines changed: 0 additions & 4 deletions

Original file line number	Diff line number	Diff line change
`@@ -620,11 +620,7 @@`
`620`	`620`	`" # create reconstructed pixels\n",`
`621`	`621`	`" x = x * y[:, :, None]\n",`
`622`	`622`	`" # flatten image into a vector shape (batch_size, vector_dim)\n",`
`623`		`-<<<<<<< HEAD`
`624`	`623`	`" flattened_x = x.contiguous().view(x.size(0), -1)\n",`
`625`		`-=======`
`626`		`- " flattened_x = x.reshape(x.size(0), -1)\n",`
`627`		`->>>>>>> 3818402479096260941d04b79061f35cc262aaa2`
`628`	`624`	`" # create reconstructed image vectors\n",`
`629`	`625`	`" reconstructions = self.linear_layers(flattened_x)\n",`
`630`	`626`	`" \n",`

Comments

(0)