Jetson Nano / minimal requirements?

[lucidrains]

Jetson Nano / minimal requirements?

[rhoen]

When I was first running the script I was seeing out of memory errors too. I was able to get it running by reducing the image size. I’ve read online elsewhere that reducing batch size can sometimes help, but I think the default batch size is already pretty low at 3, and reducing it to 1 didn’t help me. For reference my gpu has 2 gigs of ram and I was able to get running with an image size of 64.

[lucidrains]

@sjoerdju You can also adjust the network capacity! --network-capacity {num} I think the default is somewhere around 16

[lucidrains]

@sjoerdju I'll add some documentation around this soon, I wasn't expecting this much interest

[sjoerdju]

Thnx for both of your comments! I'll try and let you know if it worked.

Documentation would be fantastic. Let me know if I can be of any help. I'm rather green in this field so a script like this is great for trial and error :)

[lucidrains]

@sjoerdju yes! perhaps once you get it working, just let me know what would have helped, and I'll add it to the readme. I should definitely document the network capacity (which defines the capacity of the neural network) somewhere

[lucidrains]

added some instructions in the Readme here ef9474f