from aim.sdk.track import *
from aim.sdk.init import *
from aim.sdk.profiler import Profiler


# Serializable classes references
loss = 'Loss'
accuracy = 'Accuracy'
image = 'Image'
misclassification = 'Misclassification'
checkpoint = 'Checkpoint'
metric = 'Metric'
correlation = 'Correlation'
label_correlation = 'LabelCorrelation'
weights = 'WeightsDistribution'
gradients = 'GradientsDistribution'
hyperparams = 'HyperParameters'
stats = 'Stats'
learning_rate = 'LearningRate'
