Subject | Hash | Author | Date (UTC) |
---|---|---|---|
sha with fixed ground true sigma | b368c9b07841e462ebff4f75b260c90ce1d4571f | Thai Thien | 2020-05-01 16:55:43 |
make sha training set | d446d4ac968186bb564d907be1f34ba39c8dde5f | Thai Thien | 2020-05-01 15:56:02 |
OMP_NUM_THREADS=4 | 8e80475ba9cfd8bff7185327c81fe72b42cf0f28 | Thai Thien | 2020-04-30 19:44:28 |
ssim t6 | edf1dcc13c83aaa933d94ea77140e551c0660e14 | Thai Thien | 2020-04-30 19:41:33 |
4 threads | dc740b82dc931f9b4c971322a28876a3d95735f0 | Thai Thien | 2020-04-30 19:35:22 |
torch.set_num_threads(4) | a45027f9b9088eb55da8fa7994dd6944e572b0e2 | Thai Thien | 2020-04-30 19:33:50 |
pin_memory=False | dabbd7134c597e184c1be88831f15416579ceddc | Thai Thien | 2020-04-30 19:31:17 |
debug cpu consume | d5d9699e2fed3725cd5467cdacd1b5762f2b988f | Thai Thien | 2020-04-30 19:20:38 |
only import ssim when it is needed | 3088d6fe240f94cb47057d76e9fc4da48f2d3aaf | Thai Thien | 2020-04-30 19:16:10 |
no more batch timer | 3fea48c4642f62ab92accbe2493b6275d68c1b02 | Thai Thien | 2020-04-30 19:04:16 |
num_worker=0 | 9e21a25fd4194e99c7899f7c6b6f04d51b432c61 | Thai Thien | 2020-04-30 19:01:09 |
.to(device) | 136bafc672194946e944e85d43880f3342abf9be | Thai Thien | 2020-04-30 18:52:05 |
we load, so increase epoch | b147d3d824c0597390900cb3506495ad9d842bd4 | Thai Thien | 2020-04-30 18:45:20 |
wrong note | 31e67446689b4cecb8f118c43e6794fa437e8eb5 | Thai Thien | 2020-04-30 18:44:38 |
use ssim ccnn | 311308b98fc0a426948953ecdf8b9eec6dfd69dd | Thai Thien | 2020-04-30 18:38:25 |
loss reduce as we train, it might ok | cf1b359106f6e707e6862f2b3cbc9b5c85466ea4 | Thai Thien | 2020-04-30 18:34:19 |
replace SSIM with kornia SSIM | 1f29782fe73751182b696c18567c1de3ab84a6bc | Thai Thien | 2020-04-30 17:13:09 |
fix link | 3c55cee5cbc9fbaee0859b6ab61529029caee681 | Thai Thien | 2020-04-30 15:41:10 |
update documentation | feaaa4095d89e99f71f0684eec302d01e421708d | Thai Thien | 2020-04-30 15:35:38 |
change gpu | c7c6aa29ccee5736c2d4cb290c2ff71c6d897d57 | Thai Thien | 2020-04-30 12:31:11 |
File | Lines added | Lines deleted |
---|---|---|
train_compact_cnn.py | 2 | 3 |
train_script/CCNN/sha_fixed/ccnn_v7_t1_sha_fixed.sh | 14 | 0 |
train_script/CCNN/sha_fixed/ccnn_v7_t2_sha_fixed.sh | 14 | 0 |
File train_compact_cnn.py changed (mode: 100644) (index cd2fe73..161ef9b) | |||
... | ... | from visualize_util import get_readable_time | |
11 | 11 | import torch | import torch |
12 | 12 | from torch import nn | from torch import nn |
13 | 13 | ||
14 | |||
15 | 14 | from models import CompactCNNV2, CompactCNNV7 | from models import CompactCNNV2, CompactCNNV7 |
16 | 15 | ||
17 | 16 | import os | import os |
18 | 17 | from model_util import get_lr | from model_util import get_lr |
19 | 18 | ||
20 | 19 | COMET_ML_API = "S3mM1eMq6NumMxk2QJAXASkUM" | COMET_ML_API = "S3mM1eMq6NumMxk2QJAXASkUM" |
21 | # PROJECT_NAME = "crowd-counting-framework" | ||
22 | PROJECT_NAME = "crowd-counting-debug" | ||
20 | PROJECT_NAME = "crowd-counting-framework" | ||
21 | # PROJECT_NAME = "crowd-counting-debug" | ||
23 | 22 | ||
24 | 23 | ||
25 | 24 | def very_simple_param_count(model): | def very_simple_param_count(model): |
File train_script/CCNN/sha_fixed/ccnn_v7_t1_sha_fixed.sh added (mode: 100644) (index 0000000..e24cc9b) | |||
1 | task="ccnn_v7_t1_sha_fixed" | ||
2 | |||
3 | CUDA_VISIBLE_DEVICES=1 OMP_NUM_THREADS=5 HTTPS_PROXY="http://10.60.28.99:86" nohup python train_compact_cnn.py \ | ||
4 | --task_id $task \ | ||
5 | --note "1e-4 sha fix 15" \ | ||
6 | --model "CompactCNNV7" \ | ||
7 | --input /data/rnd/thient/thient_data/shanghaitech_with_people_density_map/ShanghaiTech_fixed_sigma/part_A \ | ||
8 | --lr 1e-4 \ | ||
9 | --decay 1e-4 \ | ||
10 | --batch_size 20 \ | ||
11 | --datasetname shanghaitech_20p \ | ||
12 | --epochs 901 > logs/$task.log & | ||
13 | |||
14 | echo logs/$task.log # for convenience |
File train_script/CCNN/sha_fixed/ccnn_v7_t2_sha_fixed.sh added (mode: 100644) (index 0000000..f233b17) | |||
1 | task="ccnn_v7_t2_sha_fixed" | ||
2 | |||
3 | CUDA_VISIBLE_DEVICES=2 OMP_NUM_THREADS=5 HTTPS_PROXY="http://10.60.28.99:86" nohup python train_compact_cnn.py \ | ||
4 | --task_id $task \ | ||
5 | --note "1e-4 sha fix 15" \ | ||
6 | --model "CompactCNNV7" \ | ||
7 | --input /data/rnd/thient/thient_data/shanghaitech_with_people_density_map/ShanghaiTech_fixed_sigma/part_A \ | ||
8 | --lr 1e-4 \ | ||
9 | --decay 1e-4 \ | ||
10 | --batch_size 20 \ | ||
11 | --datasetname shanghaitech_keepfull \ | ||
12 | --epochs 901 > logs/$task.log & | ||
13 | |||
14 | echo logs/$task.log # for convenience |