wyf-gh

Hello, I have a problem that the calculated output size is too small (512*1*0）when I train the model, is there anyone who have successfully run the pretrained train_PBAFN_stage1_fs.sh model could give me some suggestions?

13

 )

Import Error: cannot import name 'narrow_tensor_by_index' from 'torch.distributed._shard._utils'

1

### Description Thanks for your excellent work. But I have a ImportError when I run main.py. Details of the error are as follows： Traceback (most recent call last): File "/root/autodl-tmp/realfusion-main/main.py",...