Hi,
your code seems interesting and I tried to run it on a personal example.
First you don't specify the torch version, though it seems to work with the latest one.
Secondly, does this model work better with more data?
I try to run it on style of about 30 min and target of a bit more than 4 min but I receive an out of memory error in the gpu.
I tried to investigate the code, but I don't see any batch size, that I could edit so I could get your code to work in my example.
Can you propose anything?
Thanks in advance!
Hi,
your code seems interesting and I tried to run it on a personal example.
First you don't specify the torch version, though it seems to work with the latest one.
Secondly, does this model work better with more data?
I try to run it on style of about 30 min and target of a bit more than 4 min but I receive an out of memory error in the gpu.
I tried to investigate the code, but I don't see any batch size, that I could edit so I could get your code to work in my example.
Can you propose anything?
Thanks in advance!