ngoodger

#!/bin/bash

# Script for installing tmux on systems where you don't have root access.
# tmux will be installed in $HOME/local/bin.
# It's assumed that wget and a C/C++ compiler are installed.

# exit on error
set -e

TMUX_VERSION=2.5

ngoodger

import argparse

def main(input_csv, file_count):
    with open(input_csv, "r") as f:
        csv_text = f.readlines()
    header_line = csv_text[0]
    body_lines = csv_text[1:]
    line_count = len(body_lines)
    lines_per_file = int(line_count / output_file_count)
    # First file should take extra lines

ngoodger

from aiohttp import web
import argparse
import asyncio
from dataclasses import dataclass
from io import BytesIO
import numpy as np
import pyarrow
import prometheus_client as pc
import time
import torch

ngoodger

import numpy as np
import time
import torchvision
import torch
DEVICE = "cuda"
MODEL = torchvision.models.resnet18(pretrained=True).eval().to(DEVICE)
BATCH_SIZE = 64 
BATCHES = 100
x_cpu = torch.from_numpy(np.random.random((BATCH_SIZE, 3, 256, 256))
                     .astype(np.float32)).pin_memory()

ngoodger

import numpy as np
import requests
import time
import pyarrow
import torchvision
import torch
MAX_BATCH_SIZE = 1
TEST_CORRECT_OUTPUT = False
if TEST_CORRECT_OUTPUT:
    MODEL = torchvision.models.resnet18(pretrained=True).eval()

ngoodger

ngoodger

ngoodger

import gym
import math
import torch
from gym import spaces, logger
import numpy as np

class Pendulum(gym.Env):
    metadata = {
        'render.modes': ['human', 'rgb_array'],
        'video.frames_per_second': 30

ngoodger

import jax.numpy as jnp
from jax import random

class SkeletonEnv:
   
   def __init__(self):
      self.random_limit = 0.05
   
   def _get_obsv(self, state):
      return state

ngoodger

def fori_body(i, val):
  env_state, action_key, all_obsv, all_reward, all_done = val
  action = random.randint(action_key, (1,), 0, 2)[0]
  action_key = random.split(action_key)[0]
  env_state, obsv, reward, done, info = env.step(env_state, action)
  all_obsv = all_obsv.at[i].set(obsv)
  all_reward = all_reward.at[i].set(reward)
  all_done = all_done.at[i].set(done)
  val = (env_state, action_key, all_obsv, all_reward, all_done)
  return val