Is temperature a scalar

[gancx]

temperature_schedules/train_simCLR.py

Line 399 in 3ac1db6

tau = min_tau + 0.5 * (max_tau - min_tau) * (1 + torch.cos(torch.tensor(torch.pi * epoch / t_max)))

Thanks for your work. It looks like the temperature is a scalar and only related to the number of epoch. I wonder if my understanding is correct.