<filename>patch-cls/train_cls.py
# -*- coding: utf-8 -*-

import os, sys
import argparse
import torch
import torch.backends.cudnn as cudnn

from train_eng import train_thyroid


def set_args():
    parser = argparse.ArgumentParser(description='Thyroid Classification')
    parser.add_argument('--seed',            type=int,   default=1234)
    parser.add_argument('--epochs',          type=int,   default=10)
    parser.add_argument('--batch_size',      type=int,   default=32)
    # Optimization parameters
    parser.add_argument('--lr',              type=float, default=1.0e-3)
    parser.add_argument('--lr_decay_epochs', type=int,   default=2)
    parser.add_argument('--lr_decay_ratio',  type=float, default=0.3)
    parser.add_argument('--log_interval',    type=int,   default=500)
    # model directory and name
    parser.add_argument('--model_dir',       type=str,   default="../data/CV04/Models/PatchModels")
    parser.add_argument('--data_name',       type=str,   default="thyroid")
    parser.add_argument('--class_num',       type=int,   default=3)
    parser.add_argument('--model_name',      type=str,   default="vgg16bn")
    parser.add_argument('--patch_size',      type=int,   default=224)
    parser.add_argument('--session',         type=str,   default="01")
    parser.add_argument('--device_id',       type=str,   default="7",  help='which device')

    args = parser.parse_args()
    return args


if __name__ == '__main__':
    args = set_args()
    os.environ["CUDA_VISIBLE_DEVICES"] = args.device_id

    train_thyroid(args)

<filename>patch-cls/train_cls.py
# -*- coding: utf-8 -*-

import os, sys
import argparse
import torch
import torch.backends.cudnn as cudnn

from train_eng import train_thyroid


def set_args():
    parser = argparse.ArgumentParser(description='Thyroid Classification')
    parser.add_argument('--seed',            type=int,   default=1234)
    parser.add_argument('--epochs',          type=int,   default=10)
    parser.add_argument('--batch_size',      type=int,   default=32)
    # Optimization parameters
    parser.add_argument('--lr',              type=float, default=1.0e-3)
    parser.add_argument('--lr_decay_epochs', type=int,   default=2)
    parser.add_argument('--lr_decay_ratio',  type=float, default=0.3)
    parser.add_argument('--log_interval',    type=int,   default=500)
    # model directory and name
    parser.add_argument('--model_dir',       type=str,   default="../data/CV04/Models/PatchModels")
    parser.add_argument('--data_name',       type=str,   default="thyroid")
    parser.add_argument('--class_num',       type=int,   default=3)
    parser.add_argument('--model_name',      type=str,   default="vgg16bn")
    parser.add_argument('--patch_size',      type=int,   default=224)
    parser.add_argument('--session',         type=str,   default="01")
    parser.add_argument('--device_id',       type=str,   default="7",  help='which device')

    args = parser.parse_args()
    return args


if __name__ == '__main__':
    args = set_args()
    os.environ["CUDA_VISIBLE_DEVICES"] = args.device_id

    train_thyroid(args)

# -*- coding: utf-8 -*- # Optimization parameters # model directory and name