jiangweibeta / elic Goto Github PK

fine tuning时，报如下错误，是对训练集的尺寸有要求吗？
24-04-10 10:36:36.398 - INFO: Learning rate: 0.0001
Traceback (most recent call last):
File "/home/zhanghh/work/JiangWeibeta/MLIC/MLIC++/playground/train.py", line 162, in
main()
File "/home/zhanghh/work/JiangWeibeta/MLIC/MLIC++/playground/train.py", line 124, in main
current_step = train_one_epoch(
File "/home/zhanghh/work/JiangWeibeta/MLIC/MLIC++/utils/training.py", line 24, in train_one_epoch
optimizer.step()
File "/home/zhanghh/work/JiangWeibeta/MLIC/venv/lib/python3.10/site-packages/torch/optim/lr_scheduler.py", line 75, in wrapper
return wrapped(*args, **kwargs)
File "/home/zhanghh/work/JiangWeibeta/MLIC/venv/lib/python3.10/site-packages/torch/optim/optimizer.py", line 385, in wrapper
out = func(*args, **kwargs)
File "/home/zhanghh/work/JiangWeibeta/MLIC/venv/lib/python3.10/site-packages/torch/optim/optimizer.py", line 76, in _use_grad
ret = func(self, *args, **kwargs)
File "/home/zhanghh/work/JiangWeibeta/MLIC/venv/lib/python3.10/site-packages/torch/optim/adam.py", line 166, in step
adam(
File "/home/zhanghh/work/JiangWeibeta/MLIC/venv/lib/python3.10/site-packages/torch/optim/adam.py", line 316, in adam
func(params,
File "/home/zhanghh/work/JiangWeibeta/MLIC/venv/lib/python3.10/site-packages/torch/optim/adam.py", line 520, in _multi_tensor_adam
torch.foreach_lerp(device_exp_avgs, device_grads, 1 - beta1)
RuntimeError: The size of tensor a (32) must match the size of tensor b (192) at non-singleton dimension 0

On your github homepage, I saw your reproduction of the ELIC model, but I did not find your project code. If your project is open source, can you use your code? I want to learn and improve on this basis. Thank you very much for checking this issue during your busy time. I wish you a smooth and fruitful research.
在您的github主页上，我看到了您对ELIC模型的复现，但我没有找到您的项目代码。如果你的项目是开源的，你能使用你的代码吗？我想在此基础上学习和提高。非常感谢您在繁忙的时间回复这个问题。我祝愿你的研究顺利而富有成果。