tune.grid_search([False]),
"feature_folder":
tune.grid_search(["logmel_delta2_128_44k"]),
"db_path":
os.getenv("HOME") +
"/dcase/datasets/TAU-urban-acoustic-scenes-2019-mobile-development",
"model_cls":
vgg.vgg13_bn,
"model_args": {
"num_classes": 10,
"in_channels": 3
},
"composed_transform":
transforms.Compose([
# transform_utils.SelectChannel(0),
transform_utils.Normalizer()
]),
"data_set_cls":
Task1bDataSet2019,
"test_fn":
None, # no use here
"resume_model":
os.getenv("HOME") +
"/dcase/dev/ray_results/2019_diff_net_report/Trainable_0_batch_size=32,feature_folder=logmel_delta2_128_44k,lr=0.0001,mixup_alpha=0,mixup_concat_ori=False,network=vgg13_bn,opt_2020-09-28_11-57-24tndznmo5/checkpoint_170/model.pth",
},
name="2019_diff_net_report",
num_samples=1,
local_dir=os.getenv("HOME") + "/dcase/result/ray_results",
stop=TrainStopper(max_ep=30, stop_thres=30),
checkpoint_freq=1,
keep_checkpoints_num=1,
tune.grid_search(["logmel_delta2_128_44k"]),
"db_path":
os.getenv("HOME") +
"/dcase/datasets/TAU-urban-acoustic-scenes-2019-mobile-development",
"model_cls":
ResNetMod,
"model_args": {
"out_kernel_size": (132, 29)
},
"data_set_cls":
Task1bDataSet2019,
"test_fn":
None, # no use here
# "smoke_test": True,
"composed_transform":
transforms.Compose([transform_utils.Normalizer()]),
"resume_model":
os.getenv("HOME") +
"/dcase/dev/result/ray_results/2019_diff_net/Trainable_0_batch_size=16,feature_folder=logmel_delta2_128_44k,lr=0.1,mixup_alpha=0,mixup_concat_ori=False,momentum=0.9,network=re_2020-09-13_14-28-19wrlbclrp/checkpoint_143/model.pth",
},
name="2019_diff_net",
num_samples=1,
local_dir=os.getenv("HOME") + "/dcase/result/ray_results",
stop=TrainStopper(max_ep=200, stop_thres=200),
checkpoint_freq=1,
keep_checkpoints_num=1,
checkpoint_at_end=True,
checkpoint_score_attr="acc",
resources_per_trial={
"gpu": 0,
"cpu": 64