code-of-learn-deep-learning-with-pytorch/chapter9_Computer-Vision/segmentation at master · L1aoXingyu/code-of-learn-deep-learning-with-pytorch

README.md

按照 pytorch 官网安装 pytorch，将 mxtorch 下载下来，放到根目录，安装 tensorboardX 实现 tensorboard 可视化

\segmentation
	\mxtorch
	\data
	\models
	\dataset
	\checkpoints
	config.py
	main.py

打开终端，运行 bash 脚本来获取数据

bash get_data.sh

所有的配置文件都放在 config.py 里面，通过下面的代码来训练模型

python main.py train

也可以在终端修改配置，比如改变 epochs 和 batch_size

python main.py train \ 
	--max_epochs=100 \
	--batch_size=16