Put your pretrained weight under this dir
checkpoints
├── UniFuse
│ ├── UniFuse.pth
│ ├── UniFuse_st3d.pth
│ └── UniFuse_SpatialAudioGen.pth
├── BiFuseV2
│ ├── BiFuseV2.pth
│ ├── BiFuseV2_st3d.pth
│ └── BiFuseV2_SpatialAudioGen.pth
├── EGFormer
│ ├── HoHoNet.pth
│ └── HoHoNet_st3d.pth
└── HoHoNet
├── HoHoNet.pth
└── HoHoNet_st3d.pth