Code for Training
#2
by
Syoy
- opened
Hi, thanks for sharing the SSAST on HF! Do you have the code for training the AST self-supervised with HF transformers trainer and plan on sharing it as well? Or did you refer to the original code on github for training of the network?
Hi @Syoy ,
I just converted the checkpoint to HF format using the authors' GitHub repository. I found it somewhat difficult to run or reproduce the results of the paper, so I decided to convert the checkpoint to HF format first for easier comparison with my results. I plan to release the training script for the supervised AST (which is compatible to HF ecosystem) in the future though.