model_load

class aitoolbox.cloud.GoogleCloud.model_load.BaseModelGoogleStorageLoader(local_model_loader, local_model_result_folder_path='~/project/model_result', bucket_name='model-result', cloud_dir_prefix='')[source]

Bases: BaseGoogleStorageDataLoader

Base saved model loading from Google Cloud Storage

Parameters:
  • local_model_loader (AbstractLocalModelLoader) – model loader implementing the loading of the saved model for the selected deep learning framework

  • local_model_result_folder_path (str) – root local path where project folder will be created

  • bucket_name (str) – name of the bucket in the cloud storage from which the model will be downloaded

  • cloud_dir_prefix (str) – path to the folder inside the bucket where the experiments are going to be saved

class aitoolbox.cloud.GoogleCloud.model_load.PyTorchGoogleStorageModelLoader(local_model_result_folder_path='~/project/model_result', bucket_name='model-result', cloud_dir_prefix='')[source]

Bases: BaseModelGoogleStorageLoader, PyTorchS3ModelLoader

PyTorch Google Cloud Storage model downloader & loader

Parameters:
  • local_model_result_folder_path (str) – root local path where project folder will be created

  • bucket_name (str) – name of the bucket in the cloud storage from which the model will be downloaded

  • cloud_dir_prefix (str) – path to the folder inside the bucket where the experiments are going to be saved