Subject | Hash | Author | Date (UTC) |
---|---|---|---|
ready to train | dbe0d6c3271dbb22490f0877fa31ba9cd7852b99 | Thai Thien | 2020-03-04 15:55:05 |
done implement c-cnn | 2deecef953baf1e07ce5cf5477d208bc7ffa34cf | Thai Thien | 2020-03-03 17:25:03 |
fix script | 0e9d372b9ad60b32939f1e558b2a59fc7d518fa2 | Thai Thien | 2020-03-02 16:23:55 |
simple v3 to 91 epoch | 539fdd03c3e3497fd22b7db2aaa14f067cbf6f8d | Thai Thien | 2020-03-02 16:09:43 |
we train on all training data and validate on test data | 9407ef8d5b7c47c53d6f98dcb3c20208aad1d7a9 | Thai Thien | 2020-03-01 15:36:46 |
load and continue train v3 | 12421fb7330e5c9d2eed4f6e574dfe69bdfddefc | Thai Thien | 2020-03-01 14:50:01 |
add env file | b1ed02088b01af42efc8d6963b3699e0a5c31c01 | Thai Thien | 2020-03-01 11:10:56 |
sanity check dataloader | 034daa8bef69daff92891cc42b988a6c77b010f9 | Thai Thien | 2020-03-01 10:24:38 |
print train loader len | eef3995f63a631e0ec5d92e31f5d7db27fd04401 | Thai Thien | 2020-03-01 05:17:55 |
simple v3 t1 | 9ee69fce793709c3dcc692de73ddb9d01c54670e | Thai Thien | 2020-03-01 05:08:40 |
try fix load model not training | a9d6dc79a460f15bd34bc7c6a4e1359209f59f27 | Thai Thien | 2020-03-01 05:04:48 |
continue training for attn can adcrowdnet simple | 437fe77a4f50b6c6098a5251dbb42be74d8cdfc4 | Thai Thien | 2020-03-01 04:40:18 |
can use GPU 1 | 084554baa67d3eaa517cdad7be3ac3d1e8ba3e2c | Thai Thien | 2020-03-01 03:41:48 |
well, fix can train script actually | 1ff1cae8b9fc75fddc4778db804d1469713e55b2 | Thai Thien | 2020-03-01 03:25:29 |
can fix | a32e4697517e555cccaf3322250d8c298284b5aa | Thai Thien | 2020-03-01 03:24:19 |
train script | 0b9279193b68083ccb8d5072dcee333c3e0712e8 | Thai Thien | 2020-03-01 02:24:04 |
ready train can | e774b212367c0bf6b8aaa133afc3845a427a9359 | Thai Thien | 2020-03-01 02:20:35 |
simple v3 | 245472ed0682c617b1975f82c7a0c15ee3494818 | Thai Thien | 2020-02-29 17:43:39 |
simplev2 | bab195873f45b19020f6d3c5f717742c5e2becbd | Thai Thien | 2020-02-29 17:17:02 |
is to == | fbcc505592028368dfa9d6fee3e9d1b4a4e83cef | Thai Thien | 2020-02-29 16:59:40 |
File | Lines added | Lines deleted |
---|---|---|
models/__init__.py | 1 | 0 |
train_compact_cnn.py | 2 | 2 |
train_script/CCNN/ccnn_v1_t1.sh | 7 | 0 |
File models/__init__.py changed (mode: 100644) (index 4fcc014..11bd603) | |||
... | ... | from .deform_conv_v2 import DeformConv2d | |
5 | 5 | from .attn_can_adcrowdnet import AttnCanAdcrowdNet | from .attn_can_adcrowdnet import AttnCanAdcrowdNet |
6 | 6 | from .attn_can_adcrowdnet_freeze_vgg import AttnCanAdcrowdNetFreezeVgg | from .attn_can_adcrowdnet_freeze_vgg import AttnCanAdcrowdNetFreezeVgg |
7 | 7 | from .attn_can_adcrowdnet_simple import AttnCanAdcrowdNetSimpleV1, AttnCanAdcrowdNetSimpleV2, AttnCanAdcrowdNetSimpleV3 | from .attn_can_adcrowdnet_simple import AttnCanAdcrowdNetSimpleV1, AttnCanAdcrowdNetSimpleV2, AttnCanAdcrowdNetSimpleV3 |
8 | from .compact_cnn import CompactCNN |
File train_compact_cnn.py copied from file train_attn_can_adcrowdnet_simple.py (similarity 98%) (mode: 100644) (index 4096800..1876089) | |||
... | ... | from visualize_util import get_readable_time | |
9 | 9 | ||
10 | 10 | import torch | import torch |
11 | 11 | from torch import nn | from torch import nn |
12 | from models import AttnCanAdcrowdNetSimpleV3 | ||
12 | from models import CompactCNN | ||
13 | 13 | import os | import os |
14 | 14 | ||
15 | 15 | ||
... | ... | if __name__ == "__main__": | |
40 | 40 | print("len train_loader ", len(train_loader)) | print("len train_loader ", len(train_loader)) |
41 | 41 | ||
42 | 42 | # model | # model |
43 | model = AttnCanAdcrowdNetSimpleV3() | ||
43 | model = CompactCNN() | ||
44 | 44 | model = model.to(device) | model = model.to(device) |
45 | 45 | ||
46 | 46 | # loss function | # loss function |
File train_script/CCNN/ccnn_v1_t1.sh added (mode: 100644) (index 0000000..f21ad48) | |||
1 | CUDA_VISIBLE_DEVICES=5 nohup python train_compact_cnn.py \ | ||
2 | --task_id ccnn_v1_t1 \ | ||
3 | --input /data/rnd/thient/thient_data/ShanghaiTech/part_A \ | ||
4 | --lr 1e-4 \ | ||
5 | --decay 5e-4 \ | ||
6 | --datasetname shanghaitech \ | ||
7 | --epochs 33 > logs/ccnn_v1_t1.log & |