Vincentqyw
fix: roma
8b973ee
raw
history blame
1.08 kB
import argparse
BATCH_SIZE = 1
DATA_PATH = "./data/"
def get_arguments():
parser = argparse.ArgumentParser(description="training codes")
parser.add_argument("--task", type=str, help="Name of this training")
parser.add_argument(
"--data_path", type=str, default=DATA_PATH, help="Dataset root path."
)
parser.add_argument(
"--batch_size", type=int, default=BATCH_SIZE, help="Batch size for training. "
)
parser.add_argument(
"--debug_mode",
dest="debug_mode",
action="store_true",
help="If debug mode, load less data.",
)
parser.add_argument(
"--gamma",
dest="gamma",
action="store_true",
help="Use gamma compression for raw data.",
)
parser.add_argument(
"--camera",
type=str,
default="NIKON_D700",
choices=["NIKON_D700", "Canon_EOS_5D"],
help="Choose which camera to use. ",
)
parser.add_argument(
"--rgb_weight", type=float, default=1, help="Weight for rgb loss. "
)
return parser