A2J-Transformer
A2J-Transformer copied to clipboard
About code test.
Hello, I tried the installation instructions to test your code, but no matter how many Gpus I set up it will only run on Gpus 0. Also when I use one 3090 as gpu, it said cuda out of memory. But your paper said you test on a single 2080ti gpu. Looking forward to your reply!