vit-pytorch icon indicating copy to clipboard operation
vit-pytorch copied to clipboard

Pair images prediction

Open nooralahzadeh opened this issue 4 years ago • 2 comments
trafficstars

Hi, Is it possible to input two images (a pair), like Image1<sep>Image2 into the network?

nooralahzadeh avatar Dec 02 '20 12:12 nooralahzadeh

@nooralahzadeh yea of course :) you are no longer constrained to a 2 dimensional grid, the sky is the limit here

lucidrains avatar Dec 02 '20 18:12 lucidrains

@nooralahzadeh ohh, if you mean whether the current framework allows you to do that, nope it doesn't, but it's like a 5 line code change to make it do so

lucidrains avatar Dec 02 '20 18:12 lucidrains