【AttributeError: Can’t pickle local object ‘checkpoint.begin_background..bg_target’】

AttributeError: Can’t pickle local object ‘checkpoint.begin_background..bg_target’

参考github上的解决方法
1.找到begin_background()函数所在的py文件
2.注释掉定义在begin_background(self)中的def bg_target(queue):
3.将def bg_target(queue):写在begin_background(self)函数外面并添加静态方法注释 @staticmethod
4.将begin_background()中target=bg_target改为target=self.bg_target
如下

@staticmethod
    def bg_target(queue):
        while True:
            if not queue.empty():
                filename, tensor = queue.get()
                if filename is None: break
                imageio.imwrite(filename, tensor.numpy())

    def begin_background(self):
        self.queue = Queue()

        # def bg_target(queue):
        #     while True:
        #         if not queue.empty():
        #             filename, tensor = queue.get()
        #             if filename is None: break
        #             imageio.imwrite(filename, tensor.numpy())

        self.process = [
            Process(target=self.bg_target, args=(self.queue,)) \
            for _ in range(self.n_processes)
        ]

        for p in self.process: p.start()