fixed deprecated import

This commit is contained in:
brian
2024-10-19 20:09:32 +00:00
parent 14ccaace95
commit d6036a95a3
9 changed files with 37 additions and 47 deletions
+2 -2
View File
@@ -7,7 +7,7 @@ from dotenv import load_dotenv
from torchinfo import summary
from model.src.models import VisualCommunicationModel
from shared.data_store import connect, put_model
from shared.data_store import connect_minio, put_model
from shared.utils import setup_logging
DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
@@ -20,7 +20,7 @@ if __name__ == '__main__':
# setup logging
setup_logging()
# connect to minio
client = connect()
client = connect_minio()
# instantiate model
model = VisualCommunicationModel().to(DEVICE)
# show model weights