HaoxingChen commited on
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