Open zoe218 opened 3 years ago
ple 直接训练报错咋解决的啊
model = x.Model(config).to(config.device)
执行这句话的时候好慢啊 需要等5min?
Epoch [1/100]
/media/algo/data_1/project_others/202205-MTL/github/Multitask-master/train_eval_snr.py:49: UserWarning: Using a target size (torch.Size([2048])) that is different to the input size (torch.Size([1])) is deprecated. Please ensure they have the same size.
loss1 = config.loss_fn(outputs1,label1,reduction='mean')
Traceback (most recent call last):
File "run.py", line 34, in
hello,Tom 很棒的工作! 请问PLE训练的效果和论文里的一致么?我训练时差的挺多(loss没有用论文里的loss)