apm_cat_run.py

# Windows 10 Anconda python3.8
# coding=utf-8
'''
Author       : LiAo
Date         : 2022-07-14 15:23:29
LastEditTime : 2022-07-14 19:30:35
LastAuthor   : LiAo
Description  : Please add file description
'''
import argparse
import os
from src import apm_cat_train
from src import utils
os.environ["CUDA_VISIBLE_DEVICES"] = '0'

if __name__ == '__main__':
    # 设置随机数的种子,保证结果的可复现
    utils.setup_seed(123)
    parser = argparse.ArgumentParser()
    # 网络模型参数
    parser.add_argument('--backbone', type=str, default='tf_efficientnetv2_b2')
    parser.add_argument('--backbone_pretrain', type=bool, default=True)
    parser.add_argument('--pool', type=bool, default=True)
    parser.add_argument('--pool_size', type=tuple, default=(300, 300))
    parser.add_argument('--pool_type', type=str,
                        default='bicubic', help='must one of (nearest, linear, bilinear, bicubic, trilinear)')
    parser.add_argument('--weight_decay', type=float, default=1e-5)
    # 分类数目
    parser.add_argument('--num_classes', type=int, default=3)

    # 训练epoch
    parser.add_argument('--epoch', type=int, default=100)
    # 多次训练时使用
    parser.add_argument('--epoch_offset', type=int, default=0)
    # 超参数, 依据显存设置
    parser.add_argument('--batch_size', type=int,
                        default=16, help='decide by GPU RAM')
    # 学习率参数
    parser.add_argument('--lr', type=float, default=0.003)
    # used for RangeLars
    parser.add_argument('--lrf', type=float, default=0.005)
    parser.add_argument('--ann_start', type=float, default=0.5)

    # 数据集所在目录
    parser.add_argument('--dataset', type=str,
                        default='/data/liao/datasets/cqu_2021_abnormal/')
    # 权重加载路径
    parser.add_argument('--load_weight_path', type=str,
                        default=None)  # /data/liao/code/apm/result/apm/weight/best_weight.pth
    # 数据保存的路径
    parser.add_argument('--weight_path', type=str,
                        default='/data/liao/code/apm/result/apm_cat/weight/')
    parser.add_argument('--log_path', type=str,
                        default='/data/liao/code/apm/result/apm_cat/log/')
    parser.add_argument('--model', type=str, default='apm')
    parser.add_argument('--result_path', type=str,
                        default='/data/liao/code/apm/result/apm_cat/result/')

    parser.add_argument('--device', default='cuda',
                        help='device id(i.e. 0 or 0, 1 or cpu)')

    opt = parser.parse_args(args=[])

    # 执行训练和测试
    apm_cat_train.main(opt)