How to modify this to handle non-image data

[sgbaird]

Curious if you have any thoughts of how this can be applied to something other than image data. In my case, it's a task in materials informatics.

[mrkatey]

Have you considered transforming your data examples into a series of images? Sorry if that sounds too simple.

[sgbaird]

@mrkatey I've noticed that most of lucidrains's repos allow for passing tensors directly, which is great. I think it still assumes n-channel square images in many cases (not sure if image_size could take a tuple of shapes), which makes a lot of sense. I've been doing something along the lines of what you mentioned in a package called xtal2png (crystal to PNG). Still testing the performance of the representation.