e6b394f7340cbeee16bf4424cdf0367f80b60619,code/deep/finetune_AlexNet_ResNet/office31.py,,,#,20

Before Change

os.environ["CUDA_VISIBLE_DEVICES"] = "0"

DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
CUDA = True if torch.cuda.is_available() else False
LEARNING_RATE = 0.001
BATCH_SIZE_SRC = 64

After Change

import time

parser = argparse.ArgumentParser(description="Finetune")
parser.add_argument("-model", "-m", type=str, required=True, help="model name", default="alexnet")
parser.add_argument("-batchsize", "-b", type=int, required=True, help="batch size", default=64)
parser.add_argument("-cuda", "-g", type=int, required=True, help="cuda id", default=3)
args = parser.parse_args()

Instances