oneflow.nn.graph.graph_config.GraphConfig.enable_compress_memory

GraphConfig.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.

Parameters

mode (bool, optional) – [description]. Default is True.