optscale-arcee 0.1.21

ML profiling tool for OptScale

Arcee

The OptScale ML profiling tool by Hystax

Arcee is a tool that hepls you to integrate ML tasks with OptScale. This tool can automatically collect executor metadata from cloud and process stats.

Installation

Arcee requires python 3.7+ to run.

pip install optscale-arcee

Usage

First of all you need to import and init arcee in your code:

import optscale_arcee as arcee

# init arcee
arcee.init('token', 'app_name')

To use custom endpoint and enable\disable ssl checks (supports using self-signed ssl certificates):

arcee.init('token', 'test_app', endpoint_url='https://my.custom.endpoint:443/arcee/v2', ssl=False)

To send stats:

arcee.send({"loss": 2.0012, "iter": 2, "epoch": 1})

(key should be string, value - int or float, multiple values can be sent)

To add tags to application run (key, value):

arcee.tag("project", "torchvision demo")

To add milestones:

arcee.milestone("Download test data")

To add stages:

arcee.stage("calculation")

To finish model:

arcee.finish()

or for failed task:

arcee.error()