Could anyone please post a sample code of using this TTACH for binary segmentation in PyTorch? Can't seem to get it to work. Thank you all :)