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: