mirrors
/
SimCLR
mirror of https://github.com/sthalles/SimCLR.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

improved numerical computing opes in the loss function

pull/5/head
Thalles 2020-03-12 12:39:13 -03:00 committed by Thalles Silva
parent d56075a4dd
commit f78ee5d069
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -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)