def main(): args = parse_args() c = Container(account_name=args.account_name, account_key=args.account_key, container_name=args.container) if args.blob is None: print("Listing available models:") print() for blob in sorted(c.list(prefix="model-")): print(blob) else: print("Downloading {} to {}...".format(args.blob, args.model_dir)) bar = None def callback(current, total): nonlocal bar if bar is None: bar = progressbar.ProgressBar(max_value=total) bar.update(current) assert c.exists(args.blob), "model {} does not exist".format(args.blob) assert args.model_dir is not None c.get(args.model_dir, args.blob, callback=callback)
logger.log("Loaded models checkpoint at {} iterations".format( state["num_iters"])) return state if __name__ == '__main__': args = parse_args() if args.train_latent: print("Training latent") # Parse savedir and azure container. savedir = args.save_dir if args.save_azure_container is not None: account_name, account_key, container_name = args.save_azure_container.split( ":") container = Container(account_name=account_name, account_key=account_key, container_name=container_name, maybe_create=True) if savedir is None: # Careful! This will not get cleaned up. Docker spoils the developers. savedir = tempfile.TemporaryDirectory().name else: container = None # Create and seed the env. # env, monitored_env = make_env(args.env) if args.env == "MK": import imp try: map_config_file = args.map_config map_config = imp.load_source('map_config',