SATNet/options/base_options.py
BaseOptions.py
该类定义了训练和测试时使用的选项。
它还实现了一些辅助功能,如解析、打印和保存选项。
它还收集了数据集类和模型类中定义在<modify_commandline_options>函数中的其他选项。
主要用于训练和测试是定义参数设置。(这些参数设置在dataset类和model类中的<modify_commandline_options>方法中)
import argparse
import os
from util import util
import torch
import models
import data
initialize方法
定义在训练和测试中使用的通用选项。
<