Commit
·
d57e54e
1
Parent(s):
41293de
Upload 4 files
Browse files
.gitattributes
CHANGED
|
@@ -34,3 +34,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 34 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 35 |
simclr_256_e800/20221227_183052.log.json filter=lfs diff=lfs merge=lfs -text
|
| 36 |
simclr_256_e800/train_20221227_183052.log filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
| 34 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 35 |
simclr_256_e800/20221227_183052.log.json filter=lfs diff=lfs merge=lfs -text
|
| 36 |
simclr_256_e800/train_20221227_183052.log filter=lfs diff=lfs merge=lfs -text
|
| 37 |
+
moco_256_e800_t15a05A02/20230110_110029.log.json filter=lfs diff=lfs merge=lfs -text
|
| 38 |
+
moco_256_e800_t15a05A02/train_20230110_110029.log filter=lfs diff=lfs merge=lfs -text
|
moco_256_e800_t15a05A02/20230110_110029.log.json
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:79b782fbe32038e45e83b350a21bdd5a53269a85cd9d5d12779511aaf1ed1fc9
|
| 3 |
+
size 10654175
|
moco_256_e800_t15a05A02/latest.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:98dc9dda7cb9440c91a24ca22a46d47f1460435b9b3dfefb9ad0b3863d2dccf6
|
| 3 |
+
size 482065006
|
moco_256_e800_t15a05A02/mocov2_macl_resnet50_256-coslr-800e.py
ADDED
|
@@ -0,0 +1,107 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
model = dict(
|
| 2 |
+
type='MoCo',
|
| 3 |
+
queue_len=65536,
|
| 4 |
+
feat_dim=128,
|
| 5 |
+
momentum=0.999,
|
| 6 |
+
backbone=dict(
|
| 7 |
+
type='ResNet',
|
| 8 |
+
depth=50,
|
| 9 |
+
in_channels=3,
|
| 10 |
+
out_indices=[4],
|
| 11 |
+
norm_cfg=dict(type='BN')),
|
| 12 |
+
neck=dict(
|
| 13 |
+
type='MoCoV2Neck',
|
| 14 |
+
in_channels=2048,
|
| 15 |
+
hid_channels=2048,
|
| 16 |
+
out_channels=128,
|
| 17 |
+
with_avg_pool=True),
|
| 18 |
+
head=dict(type='MaclaHead', temperature=0.15))
|
| 19 |
+
data_source = 'ImageNet'
|
| 20 |
+
dataset_type = 'MultiViewDataset'
|
| 21 |
+
img_norm_cfg = dict(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
|
| 22 |
+
train_pipeline = [
|
| 23 |
+
dict(type='RandomResizedCrop', size=224, scale=(0.2, 1.0)),
|
| 24 |
+
dict(
|
| 25 |
+
type='RandomAppliedTrans',
|
| 26 |
+
transforms=[
|
| 27 |
+
dict(
|
| 28 |
+
type='ColorJitter',
|
| 29 |
+
brightness=0.4,
|
| 30 |
+
contrast=0.4,
|
| 31 |
+
saturation=0.4,
|
| 32 |
+
hue=0.1)
|
| 33 |
+
],
|
| 34 |
+
p=0.8),
|
| 35 |
+
dict(type='RandomGrayscale', p=0.2),
|
| 36 |
+
dict(type='GaussianBlur', sigma_min=0.1, sigma_max=2.0, p=0.5),
|
| 37 |
+
dict(type='RandomHorizontalFlip'),
|
| 38 |
+
dict(type='ToTensor'),
|
| 39 |
+
dict(
|
| 40 |
+
type='Normalize',
|
| 41 |
+
mean=[0.485, 0.456, 0.406],
|
| 42 |
+
std=[0.229, 0.224, 0.225])
|
| 43 |
+
]
|
| 44 |
+
prefetch = False
|
| 45 |
+
data = dict(
|
| 46 |
+
samples_per_gpu=128,
|
| 47 |
+
workers_per_gpu=8,
|
| 48 |
+
drop_last=True,
|
| 49 |
+
train=dict(
|
| 50 |
+
type='MultiViewDataset',
|
| 51 |
+
data_source=dict(
|
| 52 |
+
type='ImageNet',
|
| 53 |
+
data_prefix='./data/train',
|
| 54 |
+
ann_file='./data/train.txt'),
|
| 55 |
+
num_views=[2],
|
| 56 |
+
pipelines=[[{
|
| 57 |
+
'type': 'RandomResizedCrop',
|
| 58 |
+
'size': 224,
|
| 59 |
+
'scale': (0.2, 1.0)
|
| 60 |
+
}, {
|
| 61 |
+
'type':
|
| 62 |
+
'RandomAppliedTrans',
|
| 63 |
+
'transforms': [{
|
| 64 |
+
'type': 'ColorJitter',
|
| 65 |
+
'brightness': 0.4,
|
| 66 |
+
'contrast': 0.4,
|
| 67 |
+
'saturation': 0.4,
|
| 68 |
+
'hue': 0.1
|
| 69 |
+
}],
|
| 70 |
+
'p':
|
| 71 |
+
0.8
|
| 72 |
+
}, {
|
| 73 |
+
'type': 'RandomGrayscale',
|
| 74 |
+
'p': 0.2
|
| 75 |
+
}, {
|
| 76 |
+
'type': 'GaussianBlur',
|
| 77 |
+
'sigma_min': 0.1,
|
| 78 |
+
'sigma_max': 2.0,
|
| 79 |
+
'p': 0.5
|
| 80 |
+
}, {
|
| 81 |
+
'type': 'RandomHorizontalFlip'
|
| 82 |
+
}, {
|
| 83 |
+
'type': 'ToTensor'
|
| 84 |
+
}, {
|
| 85 |
+
'type': 'Normalize',
|
| 86 |
+
'mean': [0.485, 0.456, 0.406],
|
| 87 |
+
'std': [0.229, 0.224, 0.225]
|
| 88 |
+
}]],
|
| 89 |
+
prefetch=False))
|
| 90 |
+
optimizer = dict(type='SGD', lr=0.03, weight_decay=0.0001, momentum=0.9)
|
| 91 |
+
optimizer_config = dict()
|
| 92 |
+
lr_config = dict(policy='CosineAnnealing', min_lr=0.0)
|
| 93 |
+
runner = dict(type='EpochBasedRunner', max_epochs=800)
|
| 94 |
+
checkpoint_config = dict(interval=10, max_keep_ckpts=3)
|
| 95 |
+
log_config = dict(interval=50, hooks=[dict(type='TextLoggerHook')])
|
| 96 |
+
dist_params = dict(backend='nccl')
|
| 97 |
+
cudnn_benchmark = True
|
| 98 |
+
log_level = 'INFO'
|
| 99 |
+
load_from = None
|
| 100 |
+
resume_from = None
|
| 101 |
+
workflow = [('train', 1)]
|
| 102 |
+
persistent_workers = True
|
| 103 |
+
opencv_num_threads = 0
|
| 104 |
+
mp_start_method = 'fork'
|
| 105 |
+
work_dir = 'trained/pretrain/moco_256_e200/'
|
| 106 |
+
auto_resume = False
|
| 107 |
+
gpu_ids = range(0, 2)
|
moco_256_e800_t15a05A02/train_20230110_110029.log
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6e79687631bf36d820417177bbdff855af97129b4bfa3315bb222a4ebd6d689f
|
| 3 |
+
size 13153567
|