_fits.npy for own dataset training SPIN model

[JackieXuu]

I use my own dataset to train SPIN model and I found it need _fits.npy file. The train datasets in fits_dict.py are fixed and maybe if you can add a flag for other datasets and generate initial fits_npy file automatically?

[caizhongang]

@ZhengtianXu Thank you for the suggestion! fits_dict.py will be refactored shortly, and we will add tools to generate initial dict for SPIN (along with some other features such as feature extraction for VIBE).

Would you mind adding this request to the wishlist please?

[JackieXuu]

@ZhengtianXu Thank you for the suggestion! fits_dict.py will be refactored shortly, and we will add tools to generate initial dict for SPIN (along with some other features such as feature extraction for VIBE).

Would you mind adding this request to the wishlist please?

Yep, thanks for your help!