mito-aff-resunet-setup-54 / torch_source.json

Add PyTorch original: model_state_dict.pt + model_def.py + torch_source.json

a598102 verified 5 days ago

228 Bytes

{
 "module": "model_def.py",
 "class": "ResUNet3DValid",
 "kwargs": {
  "out_channels": 3,
  "base_channels": 40,
  "factor": 2,
  "blocks_per_level": 1
 },
 "state_dict": "model_state_dict.pt",
 "torch_version": "2.9.1+cu128"
}

	{
	"module": "model_def.py",
	"class": "ResUNet3DValid",
	"kwargs": {
	"out_channels": 3,
	"base_channels": 40,
	"factor": 2,
	"blocks_per_level": 1
	},
	"state_dict": "model_state_dict.pt",
	"torch_version": "2.9.1+cu128"
	}