Rewrite the loss functions for tree search and policy to share the same code (mostly) #5

Open pevnak opened 3 weeks ago