|
- device_target: "Ascend"
- run_distributed: False
- dataset_path: ''
- num_workers: 4
- save_checkpoint: True
- save_checkpoint_steps: 250
- keep_checkpoint_max: 100
- ckpt_save_dir: " "
- eval_checkpoint: './checkpoints/best_84641_0.7738000154495239.ckpt'
- summary_dir: "summary/"
- data_url: ''
- train_url: ''
- #"/media/serlini/data3/Datasets/GaitDatasetB-silh_test_preprocessed"
- # device_num: ""
- # hidden_dim: 256
- # dataset: "casia-b"
- # dataset_path: ""
- # pre_trained: ""
- # epoch_num: 150
- # pretrained_epoch_num: 0
- # batch_size: (8, 16)
- # shuffle_train: False
- # # conf = {
- # # "WORK_PATH": "./work",
- # # "CUDA_VISIBLE_DEVICES": "0,1,2,3",
- # # "data": {
- # # 'dataset_path': "your_dataset_path",
- # # 'resolution': '64',
- # # 'dataset': 'CASIA-B',
- # # # In CASIA-B, data of subject #5 is incomplete.
- # # # Thus, we ignore it in training.
- # # # For more detail, please refer to
- # # # function: utils.data_loader.load_data
- # # 'pid_num': 73,
- # # 'pid_shuffle': False,
- # # },
- # # "model": {
- # # 'hidden_dim': 256,
- # # 'lr': 1e-4,
- # # 'hard_or_full_trip': 'full',
- # # 'batch_size': (8, 16),
- # # 'restore_iter': 0,
- # # 'total_iter': 80000,
- # # 'margin': 0.2,
- # # 'num_workers': 3,
- # # 'frame_num': 30,
- # # 'model_name': 'GaitSet',
- # # },
- # # }
-
- # momentum: 0.9
- # loss_scale: 1024
- # weight_decay: 0.0001
- # use_label_smooth: True
- # label_smooth_factor: 0.1
- # lr: 0.1
- # lr_scheduler: "cosine_annealing"
- # eta_min: 0
- # T_max: 150
- # warmup_epochs: 5
- # checkpoint_path: ""
- # save_checkpoint: True
- # save_checkpoint_epochs: 5
- # keep_checkpoint_max: 90
- # save_checkpoint_path: "./"
- # summary_dir: "./summary_dir"
|