input_size = (3, 64, 64)
context_size = 10
class_size = 10
feature_size = 512

train_size = 4096
batch_size = 256
epoch = 12
lr_w = 0.0002
lr = 0.002
alpha = 1e4
lam = 0.1

threshold = 0.001
rounds_w = 1000