mirror of
https://github.com/storytold/FineTrainers-Conditioning.git
synced 2026-10-09 00:09:45 +00:00
sayakpaul
25 lines
581 B
Python
25 lines
581 B
Python
import torch
|
|
from accelerate import Accelerator
|
|
|
|
|
|
class State:
|
|
# Training state
|
|
seed: int = None
|
|
model_name: str = None
|
|
accelerator: Accelerator = None
|
|
weight_dtype: torch.dtype = None
|
|
train_epochs: int = None
|
|
train_steps: int = None
|
|
overwrote_max_train_steps: bool = False
|
|
num_trainable_parameters: int = 0
|
|
learning_rate: float = None
|
|
train_batch_size: int = None
|
|
generator: torch.Generator = None
|
|
num_update_steps_per_epoch: int = None
|
|
|
|
# Hub state
|
|
repo_id: str = None
|
|
|
|
# Artifacts state
|
|
output_dir: str = None
|