Exemple #1
0
                        default=8,
                        help="the kind of classes in choosing every step")

    parser.add_argument('--iter_step',
                        type=int,
                        default=20,
                        help="test acc times")
    parser.add_argument('--cls_par', type=float, default=0.3)
    parser.add_argument('--epsion', type=float, default=1e-5)
    parser.add_argument("--w1", default=0.1, type=float)
    parser.add_argument("--w3", default=0.0, type=float)
    args = parser.parse_args()

    factory = Factory()
    args.factory = factory
    args.config = factory.ConfigFactory(args.d_name)

    os.environ["CUDA_VISIBLE_DEVICES"] = args.gpu_id
    SEED = args.seed
    torch.manual_seed(SEED)
    torch.cuda.manual_seed(SEED)
    np.random.seed(SEED)
    random.seed(SEED)

    args.output_dir_src = osp.join(
        args.output, args.d_name,
        args.config["domains"][args.s_da_i][0].upper())
    if not osp.exists(osp.join(args.output_dir_src, "source_F.pt")):
        run_source(args)
    run_target(args)