def add_args()

in vis_sandbox.py [0:0]

• 54 lines of code
• 1 McCabe index (conditional complexity)

def add_args(parser):
    parser.add_argument(
        "--logdir",
        default="cachedir",
        type=str,
        help="location where log of experiments will be stored",
    )
    parser.add_argument(
        "--no-cuda", default=False, action="store_true", help="do not use GPUs for computations"
    )
    parser.add_argument(
        "--exp",
        default="transformer_gmm_uniform",
        type=str,
        help="name of experiment to run" "for pretrained model in the code, exp can be pretrained",
    )
    parser.add_argument(
        "--resume-iter",
        default=130000,
        type=int,
        help="list the iteration from which to continue training",
    )
    parser.add_argument(
        "--task",
        default="pair_atom",
        type=str,
        help="use a series of different tasks"
        "pair_atom for measuring differences between atoms"
        "sequence_recover for sequence recovery "
        "pack_rotamer for sequence recovery "
        "rotamer_trials for rotamer trials ",
    )
    parser.add_argument(
        "--pdb-name", default="6mdw.0", type=str, help="pdb on which to run analysis"
    )
    parser.add_argument(
        "--outdir", default="rotation_energies", type=str, help="output for experiments"
    )

    #############################
    ##### Analysis hyperparameters
    #############################

    parser.add_argument(
        "--rotations", default=1, type=int, help="number of rotations to use when evaluating"
    )
    parser.add_argument(
        "--sample-mode", default="rosetta", type=str, help="gmm or weighted_gauss or rosetta"
    )
    parser.add_argument(
        "--ensemble", default=1, type=int, help="number of ensembles to use for evaluation"
    )
    parser.add_argument(
        "--neg-sample",
        default=500,
        type=int,
        help="number of negative rotamer samples for rotamer trials (1-1 ratio)",
    )

    return parser