changed to cluster to test out tied weights and cross entropy