Open rimb05 opened 2 months ago
If I try to train a model with 7 stems for example, I get:
RuntimeError: The size of tensor a (2) must match the size of tensor b (7) at non-singleton dimension 1
Apollo doesn't support more than 1 stem currently.
Also it's probably won't train correctly because it requires different training procedure.
If I try to train a model with 7 stems for example, I get:
RuntimeError: The size of tensor a (2) must match the size of tensor b (7) at non-singleton dimension 1