Utils

 store_variables (pkl_fn:(<class'str'>,<class'pathlib.Path'>), size:list,
                  reorder:bool, resample:(<class'int'>,<class'list'>))

Save variable values in a pickle file.

 load_variables (pkl_fn:(<class'str'>,<class'pathlib.Path'>))

*Loads stored variable values from a pickle file.

Args: pkl_fn: File path of the pickle file to be loaded.

Returns: The deserialized value of the pickled data.*

 print_colab_gpu_info ()

Check if we have a GPU attached to the runtime.