src/similarity/siamese.py [40:61]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
parser.add_argument('--model-dir', type=str, default=os.environ['SM_MODEL_DIR'])
parser.add_argument('--output-data-dir', type=str, default=os.environ['SM_OUTPUT_DATA_DIR'])
parser.add_argument('--data-dir', type=str, default=os.environ['SM_CHANNEL_TRAIN'])

args = parser.parse_args()
    
MODEL_PATH = os.path.join(args.model_dir, 'model.pth')
MODEL_INFO_PATH = os.path.join(args.model_dir, 'model_info.pth')
CHECKPOINT_PATH = os.path.join(args.output_data_dir, 'model.pth')
CHECKPOINT_STATE_PATH = os.path.join(args.output_data_dir, 'model_info.pth')
DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

###############################################################################
# Load data
###############################################################################

TRANSFORMATIONS = \
transforms.Compose([
    transforms.Resize(224), \
    transforms.ToTensor(), \
    transforms.Normalize(mean=[0.485, 0.456, 0.406],std=[0.229, 0.224, 0.225]) \
])
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



src/similarity/siamese2.py [42:63]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
parser.add_argument('--model-dir', type=str, default=os.environ['SM_MODEL_DIR'])
parser.add_argument('--output-data-dir', type=str, default=os.environ['SM_OUTPUT_DATA_DIR'])
parser.add_argument('--data-dir', type=str, default=os.environ['SM_CHANNEL_TRAIN'])

args = parser.parse_args()
    
MODEL_PATH = os.path.join(args.model_dir, 'model.pth')
MODEL_INFO_PATH = os.path.join(args.model_dir, 'model_info.pth')
CHECKPOINT_PATH = os.path.join(args.output_data_dir, 'model.pth')
CHECKPOINT_STATE_PATH = os.path.join(args.output_data_dir, 'model_info.pth')
DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

###############################################################################
# Load data
###############################################################################

TRANSFORMATIONS = \
transforms.Compose([
    transforms.Resize(224), \
    transforms.ToTensor(), \
    transforms.Normalize(mean=[0.485, 0.456, 0.406],std=[0.229, 0.224, 0.225]) \
])
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -