45 lines
1.2 KiB
Rust
45 lines
1.2 KiB
Rust
//! Simple ResNet test without dependencies
|
|
|
|
#[cfg(test)]
|
|
mod simple_tests {
|
|
use rtx_vision::architectures::{ResNet, ResNetConfig, ResNetVariant};
|
|
use rtx_vision::{Device, Tensor};
|
|
|
|
#[test]
|
|
fn test_resnet_basic_creation() {
|
|
let device = Device::cpu();
|
|
let config = ResNetConfig::resnet18();
|
|
let result = ResNet::new(config);
|
|
assert!(result.is_ok());
|
|
}
|
|
|
|
#[test]
|
|
fn test_resnet_config_variants() {
|
|
let variants = vec![
|
|
ResNetConfig::resnet18(),
|
|
ResNetConfig::resnet34(),
|
|
ResNetConfig::resnet50(),
|
|
ResNetConfig::resnet101(),
|
|
ResNetConfig::resnet152(),
|
|
];
|
|
|
|
for config in variants {
|
|
let device = Device::cpu();
|
|
let result = ResNet::new(config);
|
|
assert!(result.is_ok());
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_resnet_forward_basic() {
|
|
let device = Device::cpu();
|
|
let config = ResNetConfig::resnet18();
|
|
let model = ResNet::new(config).unwrap();
|
|
|
|
let input = Tensor::randn(&[1, 3, 224, 224], &device).unwrap();
|
|
let result = model.forward(&input);
|
|
|
|
assert!(result.is_ok());
|
|
}
|
|
}
|