Created by: shekargoud26

This is the usage before a forward pass. The memory usage is 6831mb

This is the usage after a forward pass. memory usage is 7123mb and it doesn't decrease after forward pass.

is it possible to clear cache manually like in pytorch by calling torch.cuda.empty_cache() ?