HI @milhidaka I am getting the following error
Traceback (most recent call last):
File "../main/main.py", line 90, in
num_classes=source_train.dataset.dataset.num_class, num_domains=disc_dim, pretrained=True)
File "../util/util.py", line 72, in get_network_fn
return nets_map[name]train
File "../model/caffenet.py", line 66, in caffenet
state_dict = torch.load("alexnet_caffe.pth.tar", map_location=lambda storage, loc: storage, pickle_module=pickle)
File "/home/vinodkk/miniconda2/envs/dg/lib/python3.6/site-packages/torch/serialization.py", line 358, in load
return _load(f, map_location, pickle_module)
File "/home/vinodkk/miniconda2/envs/dg/lib/python3.6/site-packages/torch/serialization.py", line 532, in _load
magic_number = pickle_module.load(f)
_pickle.UnpicklingError: invalid load key, '<'.
PACS/default/art1
path ../../../../../dataset/result/dg_mmld/PACS/default/art1
Source domain: photo, cartoon, sketch, Target domain: art_painting
Train: 7148, Val: 795, Test: 2048
args.model caffenet