Rameshkumardas

Is there any plans to release the training script please let me know? I'm also wating for it.

``` python print('superv >>> image ok, network to') from network import TSR print('=========Model Class initialized========>>>>>>') # print("======torch.__version__=======>>>>>>>", torch.__version__) model = TSR(radius=0.87, valid_thresh=0.01, num_samples_per_ray=128, img_size=512, depth=12, embed_dim=768, num_layers=16, num_channels=1024, cross_attention_dim=768) #...

Hi @yuedajiong Please check