enable_compress_memory(mode: bool = True)¶
If true, then the graph will try its best to find the minimum memory allocation strategy. This process might take several minutes for a small graph and half an hour for a large one. The compressed memory would be closed to the lower bound of the peak memory. It benefits a lot if you need to train a lot of batches.
mode (bool, optional) – [description]. Default is True.