def main():
"""Run experiment."""
args = _parse_args()
if args.gpu < 0:
gpu_manager = GPUManager()
args.gpu = gpu_manager.get_free_gpu()
experiment_config = json.loads(args.experiment_config)
os.environ["CUDA_VISIBLE_DEVICES"] = f"{args.gpu}"
run_experiment(experiment_config, args.save, args.gpu)
def main():
"""Run experiment."""
args = _parse_args()
# Hide lines below until Lab 4
if args.gpu < 0:
gpu_manager = GPUManager()
args.gpu = gpu_manager.get_free_gpu() # Blocks until one is available
# Hide lines above until Lab 4
experiment_config = json.loads(args.experiment_config)
os.environ["CUDA_VISIBLE_DEVICES"] = f'{args.gpu}'
run_experiment(experiment_config, args.save, args.gpu)
def main():
"""Run experiment."""
args = _parse_args()
if args.gpu < 0:
gpu_manager = GPUManager()
args.gpu = gpu_manager.get_free_gpu() # Blocks until one is available
experiment_config = json.loads(args.experiment_config)
os.environ["CUDA_VISIBLE_DEVICES"] = f"{args.gpu}"
run_experiment(experiment_config,
args.save,
args.gpu,
use_wandb=not args.nowandb)
parser = argparse.ArgumentParser()
parser.add_argument("--gpu",
type=int,
default=0,
help="Provide index of GPU to use.")
parser.add_argument(
"--save",
default=False,
dest='save',
action='store_true',
help=
"If true, then final weights will be saved to canonical, version-controlled location"
)
parser.add_argument(
"experiment_config",
type=str,
help=
"JSON of experiment to run (e.g. '{\"dataset\": \"EmnistDataset\", \"model\": \"CharacterModel\", \"network\": \"mlp\"}'"
)
args = parser.parse_args()
##### Hide lines below until Lab 4
if args.gpu < 0:
gpu_manager = GPUManager()
args.gpu = gpu_manager.get_free_gpu() # Blocks until one is available
##### Hide lines above until Lab 4
experiment_config = json.loads(args.experiment_config)
os.environ["CUDA_VISIBLE_DEVICES"] = f'{args.gpu}'
run_experiment(experiment_config, args.save, args.gpu)
model.save_weights()
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument("--gpu",
type=int,
default=0,
help="Provide index of GPU to use")
parser.add_argument(
"--save",
default=False,
dest='save',
action='store_true',
help='If true, final weights are saved to verion-controlled location')
parser.add_argument("experiment_config",
type=str,
help="JSON of config for experiment")
args = parser.parse_args()
'''
GPU manager - see line 115 in https://github.com/julioadl/fsdl-text-recognizer-project/blob/master/lab5_sln/training/run_experiment.py
'''
if args.gpu < 0:
gpu_manager = GPUManager()
args.gpu = gpu_manager.get_free_gpu()
experiment_config = json.loads(args.experiment_config)
os.environ["CUDA_VISIBLE_DEVICES"] = f"{args.gpu}"
run_experiment(experiment_config, args.save, args.gpu)
