摘要: argparse模块是python用于解析命令行参数,比如python train.py --lr 0.1,这个值为0.1的lr参数只能通过argparse模块传递给python程序内部。 一般过程: import argparse parser = argparse.ArgumentParser( 阅读全文
posted @ 2018-12-05 17:00 有梦就要去实现他 阅读(495) 评论(0) 推荐(0) 编辑
摘要: 使用PyTorch设置多线程(threads)进行数据读取(DataLoader),其实是假的多线程,他是开了N个子进程(PID都连着)进行模拟多线程工作,所以你的程序跑完或者中途kill掉主进程的话,子进程的GPU显存并不会被释放,需要手动一个一个kill才行,具体方法描述如下: 1.先关闭ssh 阅读全文
posted @ 2018-12-05 12:23 有梦就要去实现他 阅读(3904) 评论(0) 推荐(0) 编辑
摘要: https://blog.csdn.net/flysky_jay/article/details/82142254 当然也可以使用top找进程,但这种方式更好 阅读全文
posted @ 2018-12-05 12:22 有梦就要去实现他 阅读(412) 评论(0) 推荐(0) 编辑
摘要: screen -S + name:创建一个名字叫做name的会话。在里面执行你想要执行的程序,再用Ctrl+a+d退出,让会话Detached,这样就能保证你的任务在后台一直运行,也不会随着终端的关闭任务就停止运行。Ctrl+d 为退出并删除会话。 screen -ls: 查看当前所有的会话。 sc 阅读全文
posted @ 2018-12-05 12:10 有梦就要去实现他 阅读(1483) 评论(0) 推荐(0) 编辑