预训练时报错

[zhaoxueyi6]

预训练时报错，网上找不到相关的问题，可否给点提示

Traceback (most recent call last):
File "/home/zxy/exp/DFMGAN-main/train.py", line 645, in
main() # pylint: disable=no-value-for-parameter
File "/home/zxy/anaconda3/envs/dftgan/lib/python3.9/site-packages/click/core.py", line 1157, in call
return self.main(*args, **kwargs)
File "/home/zxy/anaconda3/envs/dftgan/lib/python3.9/site-packages/click/core.py", line 1078, in main
rv = self.invoke(ctx)
File "/home/zxy/anaconda3/envs/dftgan/lib/python3.9/site-packages/click/core.py", line 1434, in invoke
return ctx.invoke(self.callback, **ctx.params)
File "/home/zxy/anaconda3/envs/dftgan/lib/python3.9/site-packages/click/core.py", line 783, in invoke
return __callback(*args, **kwargs)
File "/home/zxy/anaconda3/envs/dftgan/lib/python3.9/site-packages/click/decorators.py", line 33, in new_func
return f(get_current_context(), *args, **kwargs)
File "/home/zxy/exp/DFMGAN-main/train.py", line 638, in main
subprocess_fn(rank=0, args=args, temp_dir=temp_dir)
File "/home/zxy/exp/DFMGAN-main/train.py", line 473, in subprocess_fn
training_loop.training_loop(rank=rank, **args)
File "/home/zxy/exp/DFMGAN-main/training/training_loop.py", line 532, in training_loop
result_dict = metric_main.calc_metric(metric=metric, G=snapshot_data['G_ema'],
File "/home/zxy/exp/DFMGAN-main/metrics/metric_main.py", line 46, in calc_metric
results = _metric_dictmetric
File "/home/zxy/exp/DFMGAN-main/metrics/metric_main.py", line 86, in fid50k_full
fid = frechet_inception_distance.compute_fid(opts, max_real=None, num_gen=50000)
File "/home/zxy/exp/DFMGAN-main/metrics/frechet_inception_distance.py", line 29, in compute_fid
mu_gen, sigma_gen = metric_utils.compute_feature_stats_for_generator(
File "/home/zxy/exp/DFMGAN-main/metrics/metric_utils.py", line 280, in compute_feature_stats_for_generator
features = detector(images, **detector_kwargs)
File "/home/zxy/anaconda3/envs/dftgan/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1051, in _call_impl
return forward_call(*input, **kwargs)
RuntimeError: MALFORMED INPUT: lanes dont match

[Ldhlwh]

这两个issue或许有帮助
NVlabs/stylegan3#20
pytorch/pytorch#64062

[zhaoxueyi6]

感谢！我跳过了metrics，还有一个问题，在Transfer的部分报错Traceback (most recent call last):
File "/home/zxy/exp/DFMGAN-main/train.py", line 645, in
main() # pylint: disable=no-value-for-parameter
File "/home/zxy/anaconda3/envs/dftgan/lib/python3.9/site-packages/click/core.py", line 1157, in call
return self.main(*args, **kwargs)
File "/home/zxy/anaconda3/envs/dftgan/lib/python3.9/site-packages/click/core.py", line 1078, in main
rv = self.invoke(ctx)
File "/home/zxy/anaconda3/envs/dftgan/lib/python3.9/site-packages/click/core.py", line 1434, in invoke
return ctx.invoke(self.callback, **ctx.params)
File "/home/zxy/anaconda3/envs/dftgan/lib/python3.9/site-packages/click/core.py", line 783, in invoke
return __callback(*args, **kwargs)
File "/home/zxy/anaconda3/envs/dftgan/lib/python3.9/site-packages/click/decorators.py", line 33, in new_func
return f(get_current_context(), *args, **kwargs)
File "/home/zxy/exp/DFMGAN-main/train.py", line 638, in main
subprocess_fn(rank=0, args=args, temp_dir=temp_dir)
File "/home/zxy/exp/DFMGAN-main/train.py", line 473, in subprocess_fn
training_loop.training_loop(rank=rank, **args)
File "/home/zxy/exp/DFMGAN-main/training/training_loop.py", line 169, in training_loop
resume_data = legacy.load_network_pkl(f)
File "/home/zxy/exp/DFMGAN-main/legacy.py", line 21, in load_network_pkl
data = _LegacyUnpickler(f).load()
_pickle.UnpicklingError: invalid load key, '\xcf'，我试了重新下载pkl文件也不行，我想是不是pickle版本的问题，可否问一下您的pickle版本

[Ldhlwh]

我用的是python 3.8.12自带的pickle

[zhaoxueyi6]

收到，感谢回复！ 赵雪屹 ***@***.***  

…

------------------ 原始邮件 ------------------ 发件人: "Ldhlwh/DFMGAN" ***@***.***>; 发送时间: 2024年5月29日(星期三) 晚上10:20 ***@***.***>; ***@***.******@***.***>; 主题: Re: [Ldhlwh/DFMGAN] 预训练时报错 (Issue #33) 我用的是python 3.8.12自带的pickle — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: ***@***.***>