From f78ee5d069cdf6d65e8f649a7169e175facf159c Mon Sep 17 00:00:00 2001 From: Thalles Date: Thu, 12 Mar 2020 12:39:13 -0300 Subject: [PATCH] improved numerical computing opes in the loss function --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 92de58f..c11f252 100644 --- a/train.py +++ b/train.py @@ -27,7 +27,7 @@ use_cosine_similarity = config['use_cosine_similarity'] data_augment = get_data_transform_opes(s=config['s'], crop_size=96) -train_dataset = datasets.STL10('./data', split='train', download=True, transform=DataTransform(data_augment)) +train_dataset = datasets.STL10('./data', split='train+unlabeled', download=True, transform=DataTransform(data_augment)) train_loader, valid_loader = get_train_validation_data_loaders(train_dataset, config)