About use ssr_net as custom body

i wonder can i use pytorch code of ssr_net and use it as custom body
the model code is from “https://github.com/oukohou/SSR-Net-Pytorch/blob/master/SSR_models/SSR_Net_model.py