Project dependencies may have API risk issues

[PyDeps]

Hi, In Bringing-Old-Photos-Back-to-Life, inappropriate dependency versioning constraints can cause risks.

Below are the dependencies and version constraints that the project is using

torchvision
scikit-image
easydict
dill
tensorboardX
scipy
PySimpleGUI
matplotlib

The version constraint == will introduce the risk of dependency conflicts because the scope of dependencies is too strict.
The version constraint No Upper Bound and * will introduce the risk of the missing API Error because the latest version of the dependencies may remove some APIs.

After further analysis, in this project,
The version constraint of dependency PySimpleGUI can be changed to >=4.34.0,<=4.60.3.

The above modification suggestions can reduce the dependency conflicts as much as possible,
and introduce the latest version as much as possible without calling Error in the projects.

The invocation of the current project includes all the following methods.

In version PySimpleGUI-4.33.0, the API PySimpleGUI.FileBrowse whch is used by the current project in /GUI.py is missing.

The calling methods from the PySimpleGUI

PySimpleGUI.Column
PySimpleGUI.Text
PySimpleGUI.Window.read
PySimpleGUI.FileBrowse
PySimpleGUI.Window.close
PySimpleGUI.Image
PySimpleGUI.Button
PySimpleGUI.Window

The calling methods from the all methods

modify
self.fc_var
networks.GANLoss
f.read
numpy.expand_dims
model.module.optimizer_G.zero_grad
self.generate_fake
torch.nn.functional.normalize.permute
UnetGenerator.to
self.netG_A.cuda
param.numel
oneHot_size.torch.Size.torch.cuda.FloatTensor.zero_
img.convert.convert
util.visualizer.Visualizer.print_current_errors
__flip
cv2.cvtColor
y.view.permute
data.online_dataset_for_old_photos.PairOldPhotos_with_hole
matplotlib.pyplot.imshow
self.netG_A.eval
cv2.erode.astype
torch.nn.init.orthogonal_
PIL.Image.fromarray
self.up_path.append
make_dataset
pil_to_np.copy
pil_to_np
networks.Patch_Attention_4
models.networks.normalization.get_nonspade_norm_layer
options.train_options.TrainOptions
config_text.startswith
inst.cuda
image_tensor.cpu.float
torch.cuda.is_available
torch.nn.AvgPool2d
get_out_channel
label_map.size
data_transforms.convert
format
self.save_optimizer
self.criterionFeat
cv2.imencode
np_to_pil
argparse.Namespace
blocks.append
setattr
tensor_bgr.torch.Tensor.type_as.view
self.netD.forward
matplotlib.patches.Rectangle
os.chdir
name.lower
hasattr
torch.nn.L1Loss
self.netG_A.parameters
y.view.view
numpy.load
network.load_state_dict
GlobalGenerator_v2.apply
self.mapping_net.apply
cv2.imwrite
cls
torch.zeros.requires_grad_
torch.nn.functional.interpolate.repeat
os.path.splitext
models.networks.KLDLoss
x.self.phi.view
mask_expand.repeat.repeat
self.update_options_from_file.add_argument
super
base_options.BaseOptions.initialize
m.init_weights
dill.load
self.layer4
networks.get_norm_layer
self.NonLocalBlock2D_with_mask_Res.super.__init__
self.convert_visuals_to_numpy.items
torch.ByteTensor
net.cuda
blur_blending_cv2.astype
size.size.np.zeros.astype
networks.define_G
list
tensor.data.cpu
self.mapping_net.inference_forward
data.cuda
net.load_state_dict
numpy.random.normal
inst.cpu.numpy.astype
cv2.merge
self.Downsample.super.__init__
torch.pow
label_map.data.long
x.self.theta.view
StringIO.StringIO
model_name.replace.lower
wfid.write
self.conv_s
model.module.update_fixed_params
mask.astype.Image.fromarray.convert
PIL.ImageFilter.GaussianBlur
image_tensor.detach.cpu.float
input_image.numpy
irregular_hole_synthesize
base_model.BaseModel.initialize
hiddens.data.get_device
matplotlib.pyplot.gca.add_patch
model_option_setter
numpy.concatenate
skimage.transform.SimilarityTransform
torchvision.transforms.ToTensor
self.img_bytes.append
matplotlib.pyplot.figure
torch.Tensor.type_as
os.path.basename.encode
torch.autograd.Variable.append
m.weight.data.normal_
torch.Tensor
data.custom_dataset_data_loader.CustomDatasetDataLoader.initialize
torchvision.transforms.functional.to_pil_image
cv2.erode.filter
f.read.splitlines
label_tensor.cpu.float
self.img_names.append
self.load_options
dict.get
self.__getitem__
numpy.maximum
self.MultiscaleDiscriminator.super.__init__
self.downsample
result.append
run_cmd
self.netG.parameters
data.data_loader.CreateDataLoader
self.update_options_from_file.set_defaults
self.__load_bigfile
value.mean.float.mean
self.use_gpu
self.mlp_shared
torch.nn.ReplicationPad2d
networks.define_D
make_dataset_rec
min
fig.axes.get_xaxis
torch.nn.DataParallel.name
k.split
self.fc
numpy.array
pickle.dump
argparse.ArgumentTypeError
self.criterion
torch.nn.functional.tanh.view
self.G_middle_1
self.preprocess_input
self.down_sample.append
torch.__version__.startswith
html.HTML
util.tensor2im
self.compute_generator_loss
f.Image.open.convert
opt.gpu_ids.split
torch.randn
torch.sigmoid
torch.nonzero
torch.zeros.to
self.slice4
os.path.islink
detection_models.networks.UNet.eval
os.walk
__make_power_2
self.encode_input
parser.set_defaults
models.networks.architecture.SPADEResnetBlock
torchvision.transforms.Lambda
util.util.load_network
vars
exit
image_tensor.unsqueeze.size
target_cls_name.replace.lower.replace
k.model_dict.size
g_x.permute.permute
UNet_two_decoders
self.model
self.criterionFeat_feat
self.shortcut
optimizer.state_dict
torch.cat.append
networks.GlobalGenerator_DCDCv2
numpy.clip
self.slice2
self.layer1
data.base_data_loader.BaseDataLoader.initialize
network.state_dict
torch.zeros.requires_grad_.cuda
PySimpleGUI.FileBrowse
cv2.cvtColor.astype
self.opt.gpu_ids.append
self.filtered_imgs_clean.append
convertToJpeg
self.transform
scale_tensor
matplotlib.pyplot.scatter
src_r.flatten
torch.nn.InstanceNorm2d
util.visualizer.Visualizer
VGG19_torch.cuda
self.up_3
mkdir
shutil.copy
mask_unfold.torch.mean.float
labelcolormap
NameError
self.print_options
images_info.items
self.option_file_path
models.mapping_model.Pix2PixHDModel_Mapping
dlib.get_frontal_face_detector
numpy.repeat.append
torch.nn.Dropout
make_dataset.sort
image.convert.convert
data.data_loader.CreateDataLoader.load_data
torch.nn.DataParallel
self.vgg
model.module.optimizer_D.step
data.online_dataset_for_old_photos.PairOldPhotos
torch.autograd.Variable.cuda
os.listdir.sort
SummaryWriter.add_scalar
NLayerDiscriminator
modellib.__dict__.items
os.getcwd
yaml.safe_load
self.mapping_net.parameters
items.sort
range
self.update_options_from_file.parse_args
down_block
argparse.ArgumentParser.parse_args
os.getpid
any
gray_image.size
int.split
torch.nn.utils.spectral_norm.register_parameter
util.util.find_class_in_module
cluster_path.np.load.item
inp.np.transpose.astype
numpy.ones_like
t.size.self.ByteTensor.zero_.float
label_map.data.cuda
self.g
target_cls_name.replace.lower
torch.nn.init.kaiming_normal_
os.remove
self.NL_scale_3.inference_forward
util.util.natural_sort
mask.np.array.astype
mask.transpose.transpose
torch.gather
matplotlib.pyplot.title
predict_mask.data.cpu
SN
re.search.group
self.initialize_networks
B_transform
self.Tensor
value.mean.float
re.search
print_options
easydict.EasyDict
pred.size
self.initialize
predict_mask.data.cpu.float
torch.nn.SmoothL1Loss
views.index.view.expand
self.feat_D.forward
v.mean.float
torch.nn.functional.interpolate.view
self.netD
random.sample
input.numel
cv2.erode
test_image.detach
GlobalGenerator_v2
CreateDataset
self.VGG19_torch.super.__init__
torch.save
data.Load_Bigfile.BigFileMemoryLoader
filename.endswith
networks.NonLocalBlock2D_with_mask_Res
tensor2label
path.Image.open.convert
torchvision.transforms.Resize
numpy.append
image_tensor.detach.cpu
torchvision.transforms.Compose
os.path.join
text.isdigit
datetime.datetime.now
self.tf.Summary.Image
self.up_4
tensor2label.reshape
html.HTML.add_images
model.module.optimizer_G.step
img_np.np.clip.astype
loss_logger.items
self.ByteTensor
numpy.repeat
detection_models.networks.UNet
model.module.optimizer_mapping.zero_grad
torch.nn.MSELoss
torch.nn.LeakyReLU
n_label.Colorize.size
random.random
numpy.transpose.astype
feat_map.data.cuda
torch.nn.init.xavier_uniform_
B.convert.convert
self.pad
network.state_dict.items
models.networks.sync_batchnorm.SynchronizedBatchNorm2d
cv2.imread
landmark_locator
torch.utils.data.DataLoader
n_label.Colorize.cpu
model.module.optimizer_featD.step
os.path.isfile
self.slice5.add_module
dlib.shape_predictor
tensorboardX.SummaryWriter
i.y_vgg.detach
torch.nn.ModuleList
numpy.histogram
GlobalGenerator_DCDCv2
torch.cuda.set_device
face_detector
self.up_1
self.netD.parameters
ref_g.flatten
self.Mapping_Model.super.__init__
data.pix2pix_dataset.Pix2pixDataset.modify_commandline_options.add_argument
self.apply
input.detach
data.base_dataset.get_transform
torch.nn.BCELoss
self.get_target_tensor
torch.cuda.ByteTensor
html.HTML.get_image_dir
os.path.getsize
options.train_options.TrainOptions.parse
MultiscaleDiscriminator
model.module.update_learning_rate
dict
os.unlink
parser.add_argument
model_name.replace
self.Patch_Attention_4.super.__init__
tensor.size
label_map.data.long.cuda
self.netG_A.forward
super.__init__
main
numpy.transpose
v.lower
x_crop.y_crop.raw_mask.np.rint.astype
self.slice4.add_module
re.split
mask.mask.np_img.astype.Image.fromarray.convert
self.encode_z
transform_label.long
models.networks.modify_commandline_options
links.append
torch.device
input.size.self.Tensor.fill_
self.encoder
subprocess.call
os.path.dirname
hiddens.size.torch.randn.cuda
torch.nn.DataParallel.initialize
get_pad_layer
x.view.size
GlobalGenerator_v2.cuda
UNetUpBlock
t.size.torch.cuda.ByteTensor.zero_
self.parser.parse_args
self.layer5
torchvision.transforms.Scale
cls.cuda
mkdir_if_not
PySimpleGUI.Text
find_network_using_name
cog.input
pathlib.Path
models.get_option_setter
self.GlobalGenerator_DCDCv2.super.__init__
image_numpy.scipy.misc.toimage.save
find_network_using_name.modify_commandline_options
input.size
numpy.min
util.save_image
torch.nn.init.normal_
tensor2im
fig.axes.get_yaxis.set_visible
log_file.write
detection_models.networks.UNet.to
cur_url.Image.open.convert
image_tensor.unsqueeze.detach
open
numpy.clip.astype
self.get_edges
NLayerDiscriminator.apply
skimage.transform.warp
src_b.flatten
self.res_block_1
x.Image.fromarray.convert
self.load_network
random_id.self.images.clone
torch.cat
models.networks.define_E
PIL.Image.open
clslib.__dict__.items
tensor_mc.torch.Tensor.type_as.view
yaml.safe_load.items
self.netE.forward.half
inst.cpu.numpy
torch.nn.Upsample
size.size.torch.ByteTensor.fill_
x.cpu.numpy
torchvision.transforms.Normalize
PySimpleGUI.Column
torch.nn.init.constant_
errors.mean
self.fake_pool.query
mask.mask.np_img.astype
self.F_Combine
self.loader
match_histograms.astype
torch.autograd.Variable.scatter_
PySimpleGUI.Image
torch.sum
self.build_conv_block
numpy.loadtxt
torch.zeros
data.image_folder.make_dataset
f.seek
numpy.unique
os.path.abspath
torch.nn.Sequential
networks.VGGLoss_torch
self.UNetUpBlock.super.__init__
torch.nn.functional.softmax
dill.dump
SummaryWriter.add_image
torch.from_numpy
ar.transpose.transpose
cv2.copyMakeBorder
models.networks.GANLoss
img_name_len.fid.read.decode
network.cuda
models.networks.architecture.ResnetBlock
networks.ResnetBlock
calculate_lookup
argparse.ArgumentParser.add_argument
network.cpu.state_dict
t.size.torch.cuda.ByteTensor.zero_.float
self.children
self.paths_match
self.conv_0
int
cv2.GaussianBlur
models.networks.architecture.SPADEResnetBlock_non_spade
model.module.optimizer_mapping.step
inst_map.data.cuda
sets.Set.add
find_model_using_name
detection_models.sync_batchnorm.DataParallelWithCallback.conv_block
vars.items
self.down_path.append
create_network
torch.nn.Conv2d
loss_featD.backward
ValueError
get_norm_layer
cls.print_network
datetime.datetime.now.strftime
self.res_block.size
util.util.save_network
im.save
webpage.add_header
self.NL_scale_1
torch.bmm
self.theta
img.transpose
cv2.seamlessClone
opt.norm_G.replace
img.np.array.astype
Colorize
util.visualizer.Visualizer.print_save
max
torchvision.models.vgg19
opt.vars.items
norm_type.startswith
x_crop.y_crop.new_face.np.rint.astype
models.models.create_da_model
self.compute_discriminator_loss
sorted
self.save_options
data.custom_dataset_data_loader.CustomDatasetDataLoader
n_label.Colorize
detection_models.antialiasing.Downsample
n_label.Colorize.numpy
torch.nn.Linear
detection_models.networks.UNet.cpu
torch.no_grad
img.crop
torch.Size
transform_list.append
cls.init_weights
data.custom_dataset_data_loader.CustomDatasetDataLoader.name
os.path.exists
self.param_free_norm
model.module.optimizer_featD.zero_grad
Mapping_Model_with_mask_2
self.slice3.add_module
NLayerDiscriminator.cuda
x.view.view
transform_image
clean_tensorboard
data_transforms
image.tv.transforms.functional.to_pil_image.save
value.item
self.after_NL
util.visualizer.Visualizer.plot_current_errors
RuntimeError
self.BigFileMemoryLoader.super.__init__
self.res_block_2
os.listdir
img.convert.resize
tensor.torch.Tensor.type_as.view
self.fake_label_var.numel
io.BytesIO.getvalue
self.netG_B.forward
res.append
PySimpleGUI.Window
torch.nn.ReflectionPad2d
numpy.repeat.astype
data.online_dataset_for_old_photos.PairOldPhotos.initialize
__resize
webpage.get_image_dir
self.Hard_Compose
self.res_block
self.writer.add_image
self.loss
cv2.split
print
self.BaseDataset.super.__init__
m.bias.data.fill_
self.conv_img
landmark_locator.part
issubclass
self.parser.add_argument
VGG19_torch
is_image_file
all_tensor.append
numpy.clip.append
f.write
self.netG.forward
self.Mapping_Model_with_mask_2.super.__init__
blur_image_v2
ui_model.UIModel
torch.nn.functional.normalize
up
pad_type.get_pad_layer
str
self.feat_D.parameters
struct.unpack
zip
torch.nn.functional.softmax.masked_fill
html.HTML.save
get_size
io.BytesIO
torch.nn.functional.fold
mask.transpose.resize
self.reparameterize
torch.nn.functional.leaky_relu
PIL.Image.fromarray.save
t.size.torch.cuda.ByteTensor.zero_.half
reversed
self.norm_s
image_tensor.cpu.float.numpy
torch.nn.functional.unfold.permute
self.real_label_var.numel
self.update_options_from_file
views.index.view.expand.view
n_label.Colorize.max
slice
image.np.ones_like.astype
torch.clamp.cpu
self.netG_A.forward.detach
images_info.data.cpu
numpy.pad
netE_cls.modify_commandline_options.parse_known_args
torch.nn.utils.spectral_norm
self.mapping_net
mask.transpose.convert
pickle.load
pretrained_dict.items.items
matplotlib.pyplot.gca
torch.nn.ReLU
detection_models.sync_batchnorm.DataParallelWithCallback.model
x.size
float
calculate_cdf
torch.autograd.Variable.half
self.slice1
image_tensor.detach.cpu.float.numpy
compute_transformation_matrix.estimate
skimage.io.imsave
search
f.read.splitlines.append
torch.nn.functional.unfold
transform_label
data.base_dataset.get_params
type
SummaryWriter
__scale_shortside
torchvision.utils.make_grid
self.mapping_net.cuda
real_image.data.cuda
visuals.items
self.VGGLoss_torch.super.__init__
util.mkdirs
self.NL_scale_3
self.mlp_beta
os.path.basename
txts.append
webpage.add_images
img_fid.read
output_ins.torch.mean.expand_as
self.opt.gpu_ids.split
self.get_paths
P.data.cpu
self.head_0
img_new.astype
torch.sum.repeat
os.makedirs
options.vars.items
torchvision.utils.save_image
self.BaseNetwork.super.__init__
UnetSkipConnectionBlock
image_tensor.cpu
self.G_middle_0
self.compute_latent_vector_size
network.cpu
scratch_file.Image.open.convert
v.size
out.np.clip.astype
models.networks.normalization.SPADE
self.save_network
torchvision.transforms.RandomCrop
self.up_2
full_label.append
zero_mask
pix2pixHD_model_DA.Pix2PixHDModel
util.image_pool.ImagePool
data.online_dataset_for_old_photos.PairOldPhotos.name
torch.clamp
inst.cpu
enumerate
numpy.nonzero
part_url.Image.open.convert
PIL.Image.BICUBIC.transforms.Scale
self.loss_filter
self.singleD_forward
opt.gpu_ids.append
__crop
self.tf.Summary.Value
models.networks.define_D
errors.mean.float
self.NLayerDiscriminator.super.__init__
run.run_cmd
UNetConvBlock
open.write
inst.size
delattr
opt_file.write
self.Encoder.super.__init__
i.int.inst.nonzero.size
self.GANLoss.super.__init__
ref_r.flatten
self.criterionVGG
img_new.astype.Image.fromarray.convert
torch.nn.init.xavier_normal_
self.netG_B.eval
x.cpu
numpy.ones
self.ResnetBlock.super.__init__
fig.axes.get_xaxis.set_visible
time.time
self.convert_visuals_to_numpy
mask_unfold.torch.mean.float.repeat
self.netG_B.cuda
self.NL_scale_2.inference_forward
self.Mapping_Model_with_mask.super.__init__
detection_models.sync_batchnorm.DataParallelWithCallback.block
PySimpleGUI.Button
self.norm_1
i.int.b.b.inst.nonzero
self.load_optimizer
detection_models.networks.UNet.load_state_dict
self.gather_options
tensorflow.summary.FileWriter
block.append
img_summaries.append
self.update_options_from_file.parse_known_args
torchvision.transforms.CenterCrop
self.writer.add_scalar
torch.unsqueeze
PySimpleGUI.Window.read
x.self.g.view
PySimpleGUI.Window.close
PySimpleGUI.In
detection_models.sync_batchnorm.DataParallelWithCallback.center_crop
sets.Set
numpy.concatenate.append
synthesize_gaussian
index.hiddens.unsqueeze
model.module.save
self.sample_features
histogram.cumsum.max
PySimpleGUI.VSeperator
self.NL_scale_1.inference_forward
match_histograms
A_transform
self.netE.parameters
numpy.rint
model
t.size.self.ByteTensor.zero_
torch.nn.Tanh
self.netG.forward.size
img.resize
ims.append
NotImplementedError
loss_dict.get.backward
numpy.zeros
warnings.filterwarnings
self.norm_0
self.layer2
t.size
self.inference
self.netE
self.KLDLoss
torch.randn_like.mul
torch.autograd.Variable
image.cv2.imencode.tobytes
torch.load
fig.axes.get_yaxis
images.append
detection_models.sync_batchnorm.DataParallelWithCallback.up
self.parser.get_default
synthesize_salt_pepper
self.before_NL
self.criterionImage
UNet_two_decoders.to
net.parameters
self.slice1.add_module
classname.find
cv2.convertScaleAbs
fid.read
join
atoi
label_map.cuda
self.parameters
self.FloatTensor.fill_
pix2pixHD_model_DA.InferenceModel
time.gmtime
numpy.random.randint
self.criterionGAN
numpy.reshape
open.close
torch.nn.Sigmoid
make_dot.render
self.NL_scale_2
torch.nn.functional.interpolate.float
i.self.down_sample
self.netG
numpy.savetxt
random.randint
model.module.optimizer_D.zero_grad
dict.items
torch.matmul
Query_Patch.permute.permute
image_tensor.unsqueeze.dim
torch.optim.Adam
numpy.max
self.netG.detach
self.writer.add_summary
config.gpu_ids.split
self.W
synthesize_low_resolution
mask.repeat.repeat
torch.cuda.empty_cache
self.layer3
self.layer6
ntpath.basename
Mapping_Model_with_mask
ar.transpose.astype
tempfile.mkdtemp
self.writer.add_scalars
self.netE.forward
make_dot
self.NL
data.pix2pix_dataset.Pix2pixDataset.modify_commandline_options.set_defaults
models.networks.VGGLoss
self.slice3
scratch_image_scale.cpu.cpu
detection_models.sync_batchnorm.DataParallelWithCallback.last
self.discriminate
input.get_device
width.width.height.height.old_face.np.pad.np.rint.astype
torch.stack
round
numpy.random.permutation
Mapping_Model
functools.partial
i.int.inst.nonzero
self.tf.Summary
numpy.random.choice
self.register_buffer
util.util.mkdirs
self.model.clone
errors.items
PIL.Image.composite
torch.randn_like
clean_folder
A.convert.convert
data_transforms.save
net.cpu
self.init_loss_filter
ResnetBlock
norm_layer
v.mean.float.mean
scratch_image_scale.cpu.to
torchsummary.summary
importlib.import_module
_standard_face_pts
self.netG.requires_grad_
numpy.ceil
image_tensor.unsqueeze.unsqueeze
online_add_degradation_v2
self.mlp_gamma
gray_image.label.cpu
self.conv_block
struct.pack
detection_models.sync_batchnorm.DataParallelWithCallback.first
getattr
os.path.isdir
shutil.rmtree
PIL.Image.fromarray.astype
save_options
detection_models.sync_batchnorm.DataParallelWithCallback
image_path.replace
torch.index_select
n_label.Colorize.dim
isinstance
src_g.flatten
self.images.append
skimage.transform.warp.astype
time.strftime
gc.collect
__scale_width
sys.exit
PySimpleGUI.Window.update
set
loss_D.backward
compute_transformation_matrix
generated.data.cpu
optimizer.load_state_dict
self.conv_1
layer.weight.size
label.data.cuda
matplotlib.pyplot.axis
torch.nn.functional.tanh
html.HTML.add_header
matplotlib.pyplot.show
random.uniform
i.model
blur_blending_cv2
random.choice
self.decoder
torch.exp
cv2.LUT
synthesize_speckle
self.netG_B.parameters
self.up
self.actvn
cv2.boundingRect
m.reset_parameters
torch.mean
self.FloatTensor
self.up_0
torch.nn.BatchNorm2d
models.models.create_model
self.UNetConvBlock.super.__init__
self.phi
self.feat_discriminate
ref_b.flatten
histogram.cumsum
UnetGenerator
noisy.np.clip.astype
img_url.Image.open.convert
data.pix2pix_dataset.Pix2pixDataset.modify_commandline_options
y.permute.contiguous
torch.nn.functional.interpolate
torch.nn.ConvTranspose2d
k.pretrained_dict.size
layer.size
self.postprocess
data.online_dataset_for_old_photos.UnPairOldPhotos_SR
torch.cuda.FloatTensor
torch.max
self.slice5
index.self.img_bytes.io.BytesIO.Image.open.convert
j.i.pred_real.detach
self.fc_mu
torch.nn.functional.conv2d
util.tensor2label
argparse.ArgumentParser
data.face_dataset.FaceTestDataset.initialize
data.face_dataset.FaceTestDataset
self.slice2.add_module
self.divide_pred
skimage.img_as_ubyte
pred.size.append
get_landmark
len
models.networks.define_G
scipy.misc.toimage
cv2.GaussianBlur.astype
str.split
net.cpu.state_dict

Could please help me check this issue?
May I pull a request to fix it?
Thank you very much.