Switch ON/OFF programatically ?

[vsraptor]

Is there some way to Switch ON/OFF programatically the cache ?

I have a scenario where I have the following logic :

 train() - cycle
 batch predict() - cycle
 predict()

I want to be able to switch memorization ON for batch-predict() cycle, but be OFF otherwise ?