复现C_TRAN(General Multi-label Image Classification with Transformers)时,遇到的问题如下
解决方法:
在epoch前加一句
if __name__=='__main__':
调整batch_size等参数后,运行如下:
可能的原因:
多线程程序要放在主函数中训练。
参考:
pytorch使用出现"RuntimeError: An attempt has been made to start a new process before the..." 解决方法-CSDN博客
Python 中的 if __name__ == '__main__' 该如何理解 | Huoty's Blog (konghy.cn)