training

Oct 28, 2024

adab43c · Oct 28, 2024

Name	Name	Last commit message	Last commit date
parent directory ..
.gitignore	.gitignore	updated gitignore	Mar 27, 2020
color.py	color.py	updated copyright year	May 15, 2023
compare_image.py	compare_image.py	training: added support for 1- and 2-channel images	Jul 4, 2023
config.py	config.py	training: upgrade to PyTorch 2.4	Aug 3, 2024
convert_image.py	convert_image.py	training: added support for 1- and 2-channel images	Jul 4, 2023
dataset.py	dataset.py	training: cleanup	Jun 9, 2024
export.py	export.py	training: add export to ONNX	Jun 29, 2024
find_lr.py	find_lr.py	training: cleanup	Jun 9, 2024
image.py	image.py	apps, training: added support for non-standard 2-channel PFM/PHM and …	Jul 5, 2023
infer.py	infer.py	training: more efficient inference	Jun 11, 2024
loss.py	loss.py	training: update AMP usage	Jun 11, 2024
model.py	model.py	training: add unet_small model	Apr 28, 2024
preprocess.py	preprocess.py	training: more efficient inference	Jun 11, 2024
result.py	result.py	training: upgrade to PyTorch 2.4	Aug 3, 2024
split_exr.py	split_exr.py	training: split_exr: added support for subimages	Dec 13, 2023
ssim.py	ssim.py	Merge remote-tracking branch 'origin/devel' into oidn2	May 15, 2023
train.py	train.py	training: workaround for torch.compile bug	Oct 28, 2024
tza.py	tza.py	updated copyright year	May 15, 2023
util.py	util.py	training: add torch.compile support	Jun 10, 2024
visualize.py	visualize.py	updated copyright year	May 15, 2023

Provide feedback