attense

I read about the paper and try to reproduce the simulation but cuda memory always haunts me. I try to add ``` for p1 in self.push_color_trunk.parameters(): p1.requires_grad = False for...