33 lines
904 B
Python
33 lines
904 B
Python
"""Definition of function to generate a new randomly initialized model and save
|
|
it in Minio datastore."""
|
|
|
|
from pathlib import Path
|
|
|
|
from dotenv import load_dotenv
|
|
from torchinfo import summary
|
|
|
|
from model.src.models import VisualCommunicationModel
|
|
from shared.datastore import Datastore
|
|
from shared.utils import setup_logging
|
|
|
|
if __name__ == '__main__':
|
|
# load in env file
|
|
env_path = Path(__file__).parent.parent / 'server.env'
|
|
assert env_path.exists()
|
|
load_dotenv(env_path)
|
|
# setup logging
|
|
setup_logging()
|
|
# connect to minio
|
|
datastore = Datastore()
|
|
datastore.connect()
|
|
# instantiate model
|
|
model = VisualCommunicationModel(download_resnet_weights=True)
|
|
# show model weights
|
|
summary(model)
|
|
# put buffer in minio bucket
|
|
hash_str = datastore.put_model(
|
|
model=model,
|
|
)
|
|
print(f"hash string: {hash_str}")
|
|
print('saved to Minio')
|