Skip to content

Commit

Permalink
v1.2
Browse files Browse the repository at this point in the history
  • Loading branch information
xiezheng-cs committed Jul 30, 2020
1 parent 88b239e commit 054033a
Show file tree
Hide file tree
Showing 5 changed files with 132 additions and 72 deletions.
14 changes: 14 additions & 0 deletions .idea/deployment.xml

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

15 changes: 15 additions & 0 deletions .idea/webServers.xml

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

150 changes: 91 additions & 59 deletions .idea/workspace.xml

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

11 changes: 5 additions & 6 deletions hocon_config/mobilenetv2_caltech-256-30.hocon
Original file line number Diff line number Diff line change
@@ -1,16 +1,15 @@
# ------------- data options ----------------------------------------
target_dataset = "Caltech_256-30" # Stanford_Dogs, Caltech_256-30, Caltech_256-60, CUB-200-2011, Food-101
target_data_dir = "/mnt/ssd/datasets/Fine-Grained_Recognition/"
# target_data_dir = "/home/datasets/Fine-Grained_Recognition/"


# ------------- general options -------------------------------------------
outpath = "./quantized_transfer/mobilenet_v2/ours/4bits"
outpath = "./exp_log/mobilenet_v2/DTQ/4bits"
gpu_id = "7" # single-gpu
print_freq = 10
batch_size = 64
num_workers = 4
exp_id = "20200303"
exp_id = "20200730"


# ------------- common optimization options ----------------------------
Expand All @@ -27,16 +26,16 @@ base_task = "imagenet" # imagenet
base_model_name = "mobilenet_v2" # resnet50, mobilenet_v2
image_size = 224 # 224 for resnet101, mobilenet_v2
data_aug = "default" # default
bits_weights = 4 # 32, 2, 4 bit
bits_activations = 4 # 32, 2, 4 bit
bits_weights = 4 # 32, 5, 4 bit
bits_activations = 4 # 32, 5, 4 bit


# ------------- finetuning options ------------------------------------------
loss_type = "CrossEntropyLoss" # CrossEntropyLoss, loss_label_smoothing
lr_scheduler = "steplr" # coslr, steplr
reg_type = "channel_att_fea_map_learn" # channel_att_fea_map_learn, fea_loss
lambada = 0.5 # kl_loss weight, lambada =0.0, 0.5, 1.0, 2.0
theta = 0.01 # channel_att_fea_map_loss theta 0.01
theta = 0.01 # channel_att_fea_map_loss theta 0.01
T = 30.0


Expand Down
14 changes: 7 additions & 7 deletions hocon_config/resnet50_caltech-256-30.hocon
Original file line number Diff line number Diff line change
Expand Up @@ -4,12 +4,12 @@ target_data_dir = "/mnt/ssd/datasets/Fine-Grained_Recognition/"


# ------------- general options -------------------------------------------
outpath = "./quantized_transfer_log/Google_quantized_transfer_4bits_resnet50/"
gpu_id = "4,5"
outpath = "./exp_log/resnet-50/DTQ/4bits"
gpu_id = "4"
print_freq = 10
batch_size = 64
num_workers = 4
exp_id = "20200222"
exp_id = "20200730"


# ------------- common optimization options ----------------------------
Expand All @@ -24,18 +24,18 @@ gamma = 0.1
# ------------- model options ------------------------------------------
base_task = "imagenet" # imagenet
base_model_name = "resnet50" # resnet50, mobilenet_v2
image_size = 224 # 224 for resnet101, mobilenet_v2
image_size = 224 # 224 for resnet50, mobilenet_v2
data_aug = "default" # default
bits_weights = 4 # 32, 2, 4 bit
bits_activations = 4 # 32, 2, 4 bit
bits_weights = 4 # 32, 5, 4 bit
bits_activations = 4 # 32, 5, 4 bit


# ------------- finetuning options ------------------------------------------
loss_type = "CrossEntropyLoss" # CrossEntropyLoss, loss_label_smoothing
lr_scheduler = "steplr" # coslr, steplr
reg_type = "channel_att_fea_map_learn" # channel_att_fea_map_learn, fea_loss
lambada = 0.5 # kl_loss weight, lambada =0.0, 0.5, 1.0, 2.0
theta = 0.01 # channel_att_fea_map_loss theta 0.01
theta = 0.01 # channel_att_fea_map_loss theta 0.01
T = 60.0


Expand Down

0 comments on commit 054033a

Please sign in to comment.