Aim 빠른 시작 — 3단계로 로깅 시작

Aim 빠른 시작

공식 문서는 '3단계로 시작'이라고 말해요. 설치 → 로깅 → 서버·UI 실행 순서예요.

1. 설치

pip install aim

2. 프로젝트에 초기화

Run을 만들고, 하이퍼파라미터와 메트릭을 로깅해요.

from aimstack.base import Run, Metric

# Create a run
run = Run()
run['hparams'] = {
    'lr': 0.001,
    'batch_size': 32
}

# Create a metric
metric = Metric(run, name='loss', context={'epoch': 1})

for i in range(1000):
      metric.track(i, epoch=1)

3. 서버와 UI

aim server   # Aim Server
aim ui       # Aim UI

로거 구성

기본 Run 외에 저장 위치를 지정할 수도 있어요. 원격으로도 실행할 수 있고, 이때 repo는 원격 배포지가 돼요.

from aimstack.base import Run
aim_run = Run(repo='/path/to/repo/.aim')

aim_run['hparams'] = {'lr': 0.001, 'batch_size': 32}
my_metric = Metric(aim_run, name='my-metric', context={'env': 'aim-test'})
my_metric.track(0.0002)

PyTorch Lightning 통합

AimLogger를 만들어 Trainer의 logger 인자로 넘기면 돼요.

from aimstack.pytorch_lightning_tracker.loggers import BaseLogger as AimLogger
aim_logger = AimLogger(experiment='aim_on_pt_lightning',
                       train_metric_prefix='train_',
                       val_metric_prefix='val_')
trainer = Trainer(gpus=1, progress_bar_refresh_rate=20, max_epochs=5, logger=aim_logger)

더 알아보기

• Aim 빠른 시작
• Aim 문서 홈